Update app.py

app.py

@@ -41,7 +41,10 @@ def gen_fn(model_str, prompt, negative_prompt, max_retries=10):
     while retries < max_retries:
         try:
             noise = str(randint(0, 9999999))
-            result = models_load[model_str](f'{prompt} {noise}', negative_prompt=negative_prompt)
+            if hasattr(models_load[model_str], 'negative_prompt'):
+                result = models_load[model_str](f'{prompt} {noise}', negative_prompt=negative_prompt)
+            else:
+                result = models_load[model_str](f'{prompt} {noise}')
             return result
         except Exception as e:
             print(f"Error generating image: {e}")