fffiloni commited on
Commit
fefc151
·
verified ·
1 Parent(s): 117ba61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -104,7 +104,7 @@ def setup_model(prompt, model, seed, num_iterations, enable_hps, hps_w, enable_i
104
 
105
  except Exception as e:
106
  print(f"Unexpected Error: {e}")
107
- return f"Something went wrong with {model} loading", None
108
 
109
  def generate_image(setup_args, num_iterations):
110
  torch.cuda.empty_cache() # Free up cached memory
 
104
 
105
  except Exception as e:
106
  print(f"Unexpected Error: {e}")
107
+ return f"Something went wrong with {model} loading: {e}", None
108
 
109
  def generate_image(setup_args, num_iterations):
110
  torch.cuda.empty_cache() # Free up cached memory