Update app.py

app.py

@@ -42,7 +42,11 @@ def gen_fn(model_str, prompt, neg_prompt=None):
     noise = str(randint(0, 9999))
     if neg_prompt is None:
         neg_prompt = 'negative prompt'
-    return models_load[model_str](prompt=prompt, noise=noise, negative_prompt=neg_prompt)
+    model_func = models_load[model_str]
+    if hasattr(model_func, 'accepts_negative_prompt') and model_func.accepts_negative_prompt:
+        return model_func(prompt=prompt, noise=noise, negative_prompt=neg_prompt)
+    else:
+        return model_func(prompt=prompt, noise=noise)
 
 def make_me():
         with gr.Row():