Update main.py

main.py

@@ -18,8 +18,7 @@ class Item(BaseModel):
     height: int = 512
 
 def generate_image(item: Item):
-    image = client.image_to_image(
-        image=None,  # We're using text-to-image, so this is None
+    image = client.text_to_image(
         prompt=item.prompt,
         negative_prompt=item.negative_prompt,
         num_inference_steps=item.num_inference_steps,
@@ -32,14 +31,14 @@ def generate_image(item: Item):
 @app.post("/generate_image/")
 async def generate_image_api(item: Item):
     image = generate_image(item)
-
+    
     # Convert the image to a byte stream
     img_byte_arr = io.BytesIO()
     image.save(img_byte_arr, format='PNG')
     img_byte_arr.seek(0)
-
+    
     # Return the image as a streaming response
     return StreamingResponse(img_byte_arr, media_type="image/png")
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)