add: initial commit

.gitignore

@@ -158,3 +158,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+model/

Dockerfile

@@ -0,0 +1,22 @@
+FROM python:3.10.8-slim-buster
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+WORKDIR /app
+COPY . /app
+
+RUN apt-get update && apt-get install -y \
+    ffmpeg libsm6 libxext6
+RUN apt-get clean && rm -rf /tmp/* /var/tmp/*
+
+RUN python -m pip install --upgrade pip
+RUN python -m pip install -r requirements.txt
+
+ENV LC_ALL C.UTF-8 
+ENV LANG C.UTF-8
+
+# container listens to port 7860
+EXPOSE 7860
+
+# Start Gradio application
+CMD ["python", "app.py", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,20 @@
+'''
+main script for gradio application
+'''
+import argparse
+import gradio as gr
+
+# modules
+from src.core.logger import logger
+from src.interface.clip import demo_interface
+
+demo = demo_interface
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--host", default="0.0.0.0", type=str)
+    parser.add_argument(
+        "--port", help="will start gradio app on this port (if available)", default=7860, type=int)
+    args_all = parser.parse_args()
+    logger.info("Gradio Application live and running !")
+    demo.queue().launch(share=False, server_name=args_all.host, server_port=args_all.port)

requirements.txt

@@ -0,0 +1,7 @@
+gradio==3.36.1
+open-clip-torch==2.20.0
+Pillow==9.5.0
+
+--find-links https://download.pytorch.org/whl/torch_stable.html
+torch==2.0.1+cpu
+torchvision==0.15.2+cpu

src/core/logger.py

@@ -0,0 +1,8 @@
+'''
+Setting up a logger.
+'''
+import logging
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+logger.addHandler(logging.StreamHandler())

src/core/singleton.py

@@ -0,0 +1,8 @@
+class SingletonMeta(type):
+    _instances = {}
+
+    def __call__(cls, *args, **kwargs):
+        if cls not in cls._instances:
+            instance = super().__call__(*args, **kwargs)
+            cls._instances[cls] = instance
+        return cls._instances[cls]

src/interface/clip.py

@@ -0,0 +1,34 @@
+# libraries
+from typing import List, Dict
+import gradio as gr
+from PIL import Image
+
+# modules
+from src.core.logger import logger
+from src.model.clip import CLIP_Model
+
+MODEL = CLIP_Model(jit=True)
+
+def clean_text(text: str) -> List[str]:
+    """function to clean gradio input text, remove the trailing and leading spaces"""
+    return list(map(lambda x: x.strip(), text.split(",")))
+
+def clip_demo_fn(image: Image.Image, text: str) -> Dict[str, float]:
+    try:
+        logger.info("demo function invoked")
+        text = clean_text(text)
+        logger.debug("clean text: %s", text)
+        return MODEL(image, text)
+    except Exception as error_msg:
+        logger.error("Error caught: %s", error_msg)
+    finally:
+        logger.info("demo function completed")
+
+demo_interface = gr.Interface(
+    fn=clip_demo_fn,
+    inputs=[
+        gr.Image(type='pil', label="Image"),
+        "text"
+    ],
+    outputs=gr.Label()
+)