Update app.py

app.py

@@ -36,9 +36,11 @@ def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choices_plus]
 
-def gen_fn(model_str, prompt):
+def gen_fn(model_str, prompt, neg_prompt=None):
     if model_str == 'NA':
         return None
+    if neg_prompt is None:
+        neg_prompt = 'negative prompt'
     noise = str(randint(0, 9999))
     return models_load[model_str](f'{prompt} {noise} {neg_prompt}')