Update app.py

app.py

@@ -58,10 +58,11 @@ def set_text_unfinished():
 def set_text_finished():
     return gr.update(visible=True, value="<h3>Generation Finished</h3>")
 #################################################
-def get_image_path_list(folder_name):
-    image_basename_list = os.listdir(folder_name)
-    image_path_list = sorted([os.path.join(folder_name, basename) for basename in image_basename_list])
-    return image_path_list
+def get_image_path(image_path):
+    if os.path.exists(image_path):
+        return [image_path]  # Return the image path in a list
+    else:
+        raise FileNotFoundError(f"Image not found: {image_path}")
 
 #################################################
 class SpatialAttnProcessor2_0(torch.nn.Module):
@@ -752,7 +753,7 @@ with gr.Blocks(css=css) as demo:
                             "Chasing a man on the street.",
                             "At the police station.",
                             ]),
-                            "Japanese Anime","Using Ref Images",get_image_path_list('./ironman.jpeg'),768,768
+                            "Japanese Anime","Using Ref Images",get_image_path('./ironman.jpeg'),768,768
             ],
             [1,0.5,0.5,3,"a woman img, wearing a white T-shirt, blue loose hair",
                    "bad anatomy, bad hands, missing fingers, extra fingers, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, bad face, fused face, cloned face, three crus, fused feet, fused thigh, extra crus, ugly fingers, horn, cartoon, cg, 3d, unreal, animate, amputation, disconnected limbs",
@@ -762,7 +763,7 @@ with gr.Blocks(css=css) as demo:
                                 "work in the company",
                                 "Take a walk next to the company at noon",
                                 "lying in bed at night"]),
-                                "Japanese Anime",  "Using Ref Images",get_image_path_list('./yennlecun.png'),768,768
+                                "Japanese Anime",  "Using Ref Images",get_image_path('./yennlecun.png'),768,768
                 ],
                 [0,0.5,0.5,3,"a man, wearing black jacket",
                    "bad anatomy, bad hands, missing fingers, extra fingers, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, bad face, fused face, cloned face, three crus, fused feet, fused thigh, extra crus, ugly fingers, horn, cartoon, cg, 3d, unreal, animate, amputation, disconnected limbs",
@@ -773,7 +774,7 @@ with gr.Blocks(css=css) as demo:
                                 "laughing happily",
                                 "lying in bed at night"
                                 ]),
-                                "Japanese Anime","Only Using Textual Description",get_image_path_list('./ironman.jpeg'),768,768
+                                "Japanese Anime","Only Using Textual Description",get_image_path('./ironman.jpeg'),768,768
                 ],
                 [0,0.3,0.5,3,"a girl, wearing white shirt, black skirt, black tie, yellow hair",
                    "bad anatomy, bad hands, missing fingers, extra fingers, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, bad face, fused face, cloned face, three crus, fused feet, fused thigh, extra crus, ugly fingers, horn, cartoon, cg, 3d, unreal, animate, amputation, disconnected limbs",
@@ -788,7 +789,7 @@ with gr.Blocks(css=css) as demo:
                             "[NC]The brown squirrel appear.",
                             "is very happy # She is very happy to see the squirrel again",
                             "[NC]The brown squirrel takes the cracker and scampers up a tree. # She gives the squirrel cracker"]),
-                    "Japanese Anime","Only Using Textual Description",get_image_path_list('./taylor.png'),768,768
+                    "Japanese Anime","Only Using Textual Description",get_image_path('./taylor.png'),768,768
                 ]
                 ],
         inputs=[seed_, sa32_, sa64_, id_length_,  general_prompt, negative_prompt, prompt_array,style,model_type,files,G_height,G_width],