jannisborn
commited on
Commit
•
49c1726
1
Parent(s):
11b4f29
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})"
|