Yntec commited on
Commit
30ebea1
·
verified ·
1 Parent(s): 7e3b341

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -80,6 +80,7 @@ def gen_fnseed(model_str, prompt, seed=1):
80
  except (Exception, asyncio.CancelledError) as e:
81
  print(e)
82
  print(f"Task aborted: {model_str}")
 
83
  result = None
84
  with lock:
85
  image = "https://huggingface.co/spaces/Yntec/ToyWorld/resolve/main/error.png"
 
80
  except (Exception, asyncio.CancelledError) as e:
81
  print(e)
82
  print(f"Task aborted: {model_str}")
83
+ raise gr.Error(f"{response.status_code}: The model failed to produce an image. Please try again.")
84
  result = None
85
  with lock:
86
  image = "https://huggingface.co/spaces/Yntec/ToyWorld/resolve/main/error.png"