jannisborn commited on
Commit
49c1726
1 Parent(s): 11b4f29
Files changed (1) hide show
  1. app.py +1 -1
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"{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})"
 
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})"