Update app.py

app.py

@@ -17,11 +17,9 @@ API_URL_IMG = "https://api-inference.huggingface.co/models/stablediffusionapi/re
 api_token = os.environ.get("api_token")
 headers = {"Authorization": f"Bearer {api_token}"}
 
-def blip_query(filename):
-    with open(filename,'rb') as f:
-        data = f.read()
-    response = requests.post(API_URL_BLIP, headers=headers, data=data)
-    return response.json()
+def blip_query(image_bytes):
+        response = requests.post(API_URL_BLIP, headers=headers, data=image_bytes)
+        return response.json()
 
 def generate_answer(prompt):
     resp = Client.create_completion("gpt3", prompt)
@@ -32,7 +30,11 @@ def generate_image(payload):
 	return response.content
 
 if inpt and file is not None:
-    imgp = blip_query(file.read())
+    image = Image.open(file)
+    image_bytes = io.BytesIO()
+    image.save(image_bytes, format=image.format)
+    image_bytes = image_bytes.getvalue()
+    imgp = query(image_bytes)
     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")