Spaces:
Sleeping
Sleeping
fix model_name wrong variable
Browse files
app.py
CHANGED
|
@@ -82,7 +82,7 @@ def infer(use_custom_model, model_name, custom_lora_weight, image_in, prompt, ne
|
|
| 82 |
|
| 83 |
if use_custom_model:
|
| 84 |
|
| 85 |
-
if
|
| 86 |
raise gr.Error("you forgot to set a custom model name.")
|
| 87 |
|
| 88 |
custom_model = model_name
|
|
|
|
| 82 |
|
| 83 |
if use_custom_model:
|
| 84 |
|
| 85 |
+
if model_name == "":
|
| 86 |
raise gr.Error("you forgot to set a custom model name.")
|
| 87 |
|
| 88 |
custom_model = model_name
|