Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|