Update app.py
Browse files
app.py
CHANGED
@@ -189,11 +189,13 @@ if default_image:
|
|
189 |
if isinstance(default_image, list):
|
190 |
# Filter out non-existent paths
|
191 |
existing_images = [img for img in default_image if os.path.exists(img)]
|
|
|
192 |
if existing_images:
|
193 |
default_image = random.choice(existing_images)
|
194 |
else:
|
195 |
default_image = None
|
196 |
elif not os.path.exists(default_image):
|
|
|
197 |
default_image = None
|
198 |
else:
|
199 |
default_image = None
|
@@ -221,6 +223,6 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
|
|
221 |
random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
|
222 |
run_button.click(generate, inputs=[prompt], outputs=[result], js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
|
223 |
result.change(fn=lambda x:x, inputs=[prompt,result], outputs=[], js=f'''(p,img)=>window.uploadImage(p, img,"process_finished","demo_hf_{cfg.get("name")}_card", "finish")''')
|
224 |
-
demo.load(fn=lambda x:x, inputs=[gr.Image(value=default_image
|
225 |
if __name__ == "__main__":
|
226 |
demo.queue().launch(show_api=False)
|
|
|
189 |
if isinstance(default_image, list):
|
190 |
# Filter out non-existent paths
|
191 |
existing_images = [img for img in default_image if os.path.exists(img)]
|
192 |
+
print(f"found cover files: {existing_images}")
|
193 |
if existing_images:
|
194 |
default_image = random.choice(existing_images)
|
195 |
else:
|
196 |
default_image = None
|
197 |
elif not os.path.exists(default_image):
|
198 |
+
print(f"cover file not existed, {default_image}")
|
199 |
default_image = None
|
200 |
else:
|
201 |
default_image = None
|
|
|
223 |
random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
|
224 |
run_button.click(generate, inputs=[prompt], outputs=[result], js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
|
225 |
result.change(fn=lambda x:x, inputs=[prompt,result], outputs=[], js=f'''(p,img)=>window.uploadImage(p, img,"process_finished","demo_hf_{cfg.get("name")}_card", "finish")''')
|
226 |
+
demo.load(fn=lambda x:x, inputs=[gr.Image(value=default_image)], outputs=[result], js='''(img)=>checkDomain(img, "huggingface")''')
|
227 |
if __name__ == "__main__":
|
228 |
demo.queue().launch(show_api=False)
|