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
|