Update app.py

app.py

@@ -1,44 +1,54 @@
 import os
 import io
-#from PIL import Image
 import base64
 
 hf_api_key = os.environ['HF_API_KEY']
 
 # Helper functions
-import requests, json
+import requests, js
+
+#Image-to-text endpoint
+
+
+def image_to_base64_str(pil_image):
+    byte_arr = io.BytesIO()
+    pil_image.save(byte_arr, format='PNG')
+    byte_arr = byte_arr.getvalue()
+    return str(base64.b64encode(byte_arr).decode('utf-8'))
 
 #Image-to-text endpoint
 def get_completion(inputs, parameters = None, ENDPOINT_URL = os.environ['HF_API_ITT_BASE']): 
+    
     headers = {
       "Authorization": f"Bearer {hf_api_key}",
       "Content-Type": "application/json"
     }
-    data = { "inputs": inputs }
+    
+    data = { "inputs" : inputs }
+    
     if parameters is not None:
-        data.update({"parameters": parameters})
-    response = requests.request("POST",
-                                ENDPOINT_URL,
-                                headers=headers,
-                                data=json.dumps(data))
+        data.update({"parameters" : parameters})
+    
+    response = requests.post(ENDPOINT_URL,
+                             headers = headers,
+                             data = json.dumps(data))
+    
     return json.loads(response.content.decode("utf-8"))
 
+
 # Gradio App
 
 import gradio as gr 
-
-def image_to_base64_str(pil_image):
-    byte_arr = io.BytesIO()
-    pil_image.save(byte_arr, format='PNG')
-    byte_arr = byte_arr.getvalue()
-    return str(base64.b64encode(byte_arr).decode('utf-8'))
+from PIL import Image
 
 def captioner(image):
-    base64_image = image_to_base64_str(image)
+    pil_image = Image.open(image)
+    base64_image = image_to_base64_str(pil_image)
     result = get_completion(base64_image)
     return result[0]['generated_text']
 
 gr.close_all()
+
 demo = gr.Interface(fn = captioner,
                     inputs = [gr.Image(label="Upload image", type="pil")],
                     outputs = [gr.Textbox(label="Caption")],
@@ -48,5 +58,4 @@ demo = gr.Interface(fn = captioner,
                     examples = ["tank-aerial-view.jpg"]
                    )
 
-demo.launch()
-
+demo.launch()