update
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ def run_inference(model_type: str, prompt: str):
|
|
| 22 |
if prompt == "":
|
| 23 |
config = eval(f"{model_type}()")
|
| 24 |
else:
|
| 25 |
-
config = eval(f
|
| 26 |
if config.modality != "token2image" and prompt != "":
|
| 27 |
raise ValueError(
|
| 28 |
f"{model_type} is an unconditional generative model, please remove prompt (not={prompt})"
|
|
|
|
| 22 |
if prompt == "":
|
| 23 |
config = eval(f"{model_type}()")
|
| 24 |
else:
|
| 25 |
+
config = eval(f'{model_type}(prompt="{prompt}")')
|
| 26 |
if config.modality != "token2image" and prompt != "":
|
| 27 |
raise ValueError(
|
| 28 |
f"{model_type} is an unconditional generative model, please remove prompt (not={prompt})"
|