Update app.py

app.py

@@ -32,7 +32,7 @@ def generate_image(payload):
 	return response.content
 
 if inpt and file is not None:
-    imgp = blip_query(file)
+    imgp = blip_query(file.read())
     pp = "Generate answer on this question: " + inpt + ". Use this image description to give answer: " + imgp[0]['generated_text']
     output = generate_answer(pp)
     aimsg = msgs.chat_message("Assistant")
@@ -44,8 +44,8 @@ elif inpt and file is None:
     aimsg.write(output)
 
 elif inpt and ("Generate" in inpt or "Draw" in inpt or "Imagine" in inpt):
-    prompt = inpt.split("--")[1]
-    output = generate_image(prompt)
+    payload = {"prompts": prompt}
+    output = generate_image(payload)
     if output:
         aimsg = msgs.chat_message("Assistant")
         aimsg.write("Image generated successfully!")