nsfwalex commited on
Commit
71001b6
1 Parent(s): aee742b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -281,7 +281,7 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
281
  else:
282
  default_image = None
283
  print(f"load_demo, urlparams={params},cover={default_image},domain={current_domain},iframe={iframe_parent_domain}")
284
- if params.get("e", "0") == "1":
285
  #update the image
286
  #bind events
287
  return [Image.open(default_image)]
 
281
  else:
282
  default_image = None
283
  print(f"load_demo, urlparams={params},cover={default_image},domain={current_domain},iframe={iframe_parent_domain}")
284
+ if params.get("e", "0") == "1" or "nsfwais.io" in current_domain or "nsfwais.io" in iframe_parent_domain or "127.0.0.1" in current_domain or "127.0.0.1" in iframe_parent_domain:
285
  #update the image
286
  #bind events
287
  return [Image.open(default_image)]