Spaces:
Running
on
A10G
Running
on
A10G
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
|