Update app.py

app.py

@@ -38,8 +38,6 @@ def gen_fn(model_str, prompt, negative_prompt=None, noise=None, cfg_scale=None,
         try:
             # Construct the function call parameters dynamically
             call_params = [full_prompt]
-            if negative_prompt:
-                call_params.append(negative_prompt)
             if cfg_scale is not None:
                 # Check if cfg_scale can be set
                 if 'cfg_scale' not in models_load[model_str].parameters:
@@ -51,7 +49,7 @@ def gen_fn(model_str, prompt, negative_prompt=None, noise=None, cfg_scale=None,
                     return None, 'num_inference_steps parameter is not supported by the model'
                 call_params.append(num_inference_steps)
             
-            result = models_load[model_str](*call_params)
+            result = models_load[model_str](*call_params, negative_prompt=negative_prompt)
             
             # Check if result is an image or a file path
             if isinstance(result, str):  # Assuming result might be a file path