Update app.py

app.py

@@ -31,6 +31,7 @@ def generate_image(prompt, negative_prompt=None, height=512, width=512, model="s
         # Generate the image using Hugging Face's inference API with additional parameters
         image = client.text_to_image(
             prompt=prompt, 
+            negative_prompt=negative_prompt, 
             height=height, 
             width=width, 
             model=model,
@@ -50,6 +51,7 @@ def generate_api():
 
     # Extract required fields from the request
     prompt = data.get('prompt', '')
+    negative_prompt = data.get('negative_prompt', None)
     height = data.get('height', 1024)  # Default height
     width = data.get('width', 720)  # Default width
     num_inference_steps = data.get('num_inference_steps', 50)  # Default number of inference steps