nsfwalex commited on
Commit
1f83221
1 Parent(s): 9c7c0d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -199,10 +199,12 @@ if default_image:
199
  default_image = None
200
  else:
201
  default_image = None
202
-
203
- with gr.Blocks(css=css,head=js,fill_height=True) as demo:
204
  url_params = gr.State()
205
- print(url_params)
 
 
206
  with gr.Row(equal_height=False):
207
  with gr.Group():
208
  result = gr.Gallery(
@@ -225,6 +227,6 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
225
  random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
226
  run_button.click(generate, inputs=[prompt], outputs=[result], js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
227
  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")''')
228
- demo.load(fn=lambda x:x, inputs=[gr.Image(value=default_image)], outputs=[result], js='''(img)=>checkDomain(img, "huggingface")''')
229
  if __name__ == "__main__":
230
  demo.queue().launch(show_api=False)
 
199
  default_image = None
200
  else:
201
  default_image = None
202
+
203
+ def on_demo_load():
204
  url_params = gr.State()
205
+ print("url params", url_params)
206
+
207
+ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
208
  with gr.Row(equal_height=False):
209
  with gr.Group():
210
  result = gr.Gallery(
 
227
  random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
228
  run_button.click(generate, inputs=[prompt], outputs=[result], js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
229
  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")''')
230
+ demo.load(fn=on_demo_load, inputs=[], outputs=[])#, js='''(img)=>checkDomain(img, "huggingface")''')
231
  if __name__ == "__main__":
232
  demo.queue().launch(show_api=False)