Update app.py

app.py

@@ -25,7 +25,7 @@ device = "cuda"
 script_directory = os.path.dirname(os.path.realpath(__file__))
 
 # download ConsistentID checkpoint to cache
-base_model_path = "SG161222/RealVisXL_V5.0"
+base_model_path = "SG161222/RealVisXL_V3.0"
 consistentID_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="ConsistentID_SDXL-v1.bin", repo_type="model")
 
 ### Load base model
@@ -82,8 +82,7 @@ def process(selected_template_images,costum_image,prompt
     if model_selected_tab==0:
         select_images = load_image(Image.open(selected_template_images))
     else:
-        # select_images = load_image(Image.fromarray(costum_image))
-        select_images = load_image(costum_image)
+        select_images = load_image(Image.fromarray(costum_image))
 
     if prompt_selected_tab==0:
         prompt = prompt_selected
@@ -144,7 +143,7 @@ def process(selected_template_images,costum_image,prompt
         width=width,    
         height=height,
         input_id_images=select_images,
-        input_image_path=selected_template_images,
+        input_image_path=selected_template_images, ### path maybe not right, do not use
         negative_prompt=negative_prompt,
         num_images_per_prompt=1,
         num_inference_steps=num_steps,