Update app.py

app.py

@@ -95,8 +95,8 @@ def query(payload, max_retries=5):
     return None
     
 # Function for direct prompt to image generation
-def generate_image_from_prompt(prompt):
-     image_bytes = query({"inputs": prompt})
+def generate_image(prompt):
+    image_bytes = query({"inputs": prompt})
 
     if image_bytes is None:
         error_img = Image.new('RGB', (300, 300), color=(255, 0, 0))