Upload folder using huggingface_hub

.gitignore

@@ -0,0 +1,4 @@
+.env
+.idea
+env
+.log

README.md

@@ -1,12 +1,7 @@
 ---
 title: ImageSense
-emoji: 🏆
-colorFrom: pink
-colorTo: pink
+app_file: image_caption/gradio_interface.py
 sdk: gradio
 sdk_version: 3.39.0
-app_file: app.py
-pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# sales-demo

image_caption/__init__.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+
+"""__init__.py: """
+
+__author__ = "Rishabh Gupta"
+__copyright__ = "Copyright 2023, Zuma"
+__created_date__ = "04-08-2023"
+
+if __name__ == '__main__':
+    pass

image_caption/get_caption.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+
+"""get_caption.py: """
+
+__author__ = "Rishabh Gupta"
+__copyright__ = "Copyright 2023, Zuma"
+__created_date__ = "04-08-2023"
+
+
+import os
+import json
+import requests
+import base64
+from io import BytesIO
+
+def get_image_caption(img_obj, prompt=None):
+    """
+
+    Args:
+        img_obj: Base 64 image object
+        prompt: Prompt to be used for captioning
+
+    Returns:
+        Captions for the image
+    """
+    # print("Image Obj", img_obj)
+
+
+    buffered = BytesIO()
+    img_obj.save(buffered, format="JPEG")
+    img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
+    url = os.environ.get("IMAGE_CAPTION_ENDPOINT")
+    auth_token = os.environ.get("IMAGE_CAPTION_AUTH_TOKEN")
+    # print("img_str", img_str)
+    payload = json.dumps({
+        "inputs": img_str,
+        "prompt": None  # Prompts are disabled as accuracy is not very good or not experimented enough.
+    })
+    headers = {
+        'Authorization': auth_token,
+        'Content-Type': 'application/json'
+    }
+    response = requests.request("POST", url, headers=headers, data=payload)
+    print("Response Status Code = ", response.status_code)
+    print("Response = ", response.text)
+    if response.status_code == 200:
+        caption = json.loads(response.text)["captions"]
+        print(f"Caption = {caption}")
+        return caption
+    else:
+        return "Error in generating the caption!"
+
+
+if __name__ == '__main__':
+    with open("/Users/rishabh/Downloads/5dbadb934042f1.16205085439.jpg", "rb") as image_file:
+        encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
+    print(get_image_caption(encoded_string))

image_caption/gradio_interface.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python
+
+"""gradio_interface.py: """
+
+__author__ = "Rishabh Gupta"
+__copyright__ = "Copyright 2023, Zuma"
+__created_date__ = "04-08-2023"
+
+import gradio as gr
+from image_caption import get_caption
+
+
+gr.Interface(fn=get_caption.get_image_caption,
+             inputs=gr.Image(type="pil"),
+             outputs=gr.Label(num_top_classes=3),
+             examples=["bathroom.jpg", "image_caption/examples/livingRoom.jpg"]).launch()
+
+if __name__ == '__main__':
+    pass

requirementx.txt

@@ -0,0 +1 @@
+gradio==-3.39.0