nsfwalex commited on
Commit
0ce2aa0
1 Parent(s): 00103f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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,visible=False)], outputs=[result], js='''(img)=>checkDomain(img, "huggingface")''')
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)