nsfwalex commited on
Commit
4400f0c
·
verified ·
1 Parent(s): ff4ece2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -222,16 +222,18 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
222
 
223
  def on_demo_load(request: gr.Request):
224
  params = dict(request.query_params)
225
- print("url params", params)
226
  if params.get("e", "0") == "1":
227
  #update the image
228
- result.update(value=[default_image])
229
  #bind events
230
  random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
 
 
 
231
 
232
  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")''')
233
  run_button.click(generate, inputs=[prompt], outputs=[result], js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
234
 
235
- demo.load(fn=on_demo_load, inputs=[], outputs=[])#, js='''(img)=>checkDomain(img, "huggingface")''')
236
  if __name__ == "__main__":
237
  demo.queue().launch(show_api=False)
 
222
 
223
  def on_demo_load(request: gr.Request):
224
  params = dict(request.query_params)
225
+ print("load_demo, url params", params, "host", request.client.host)
226
  if params.get("e", "0") == "1":
227
  #update the image
 
228
  #bind events
229
  random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
230
+ return Image.open(default_image)
231
+ return None
232
+
233
 
234
  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")''')
235
  run_button.click(generate, inputs=[prompt], outputs=[result], js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
236
 
237
+ demo.load(fn=on_demo_load, inputs=[], outputs=[result])#, js='''(img)=>checkDomain(img, "huggingface")''')
238
  if __name__ == "__main__":
239
  demo.queue().launch(show_api=False)