made changes in labels and input

app.py

@@ -1,12 +1,25 @@
+from turtle import title
 import gradio as gr
+from transformers import pipeline
+import numpy as np
+from PIL import Image
 
 
-def image_mod(image):
-    return image.rotate(45)
+pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch32")
+images="dog.jpg"
 
+def shot(image, labels_text="signature,profile picture"):
+    PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
+    labels = labels_text.split(",")
+    res = pipe(images=PIL_image, 
+           candidate_labels=labels,
+           hypothesis_template= "This is a photo of a {}")
+    return {dic["label"]: dic["score"] for dic in res}
+    
+iface = gr.Interface(shot, 
+                    "image", 
+                    "label", 
+                    description="Add a picture and a list of labels separated by commas",
+                    title="Zero-shot Image Classification")
 
-iface = gr.Interface(image_mod, gr.inputs.Image(type="pil"), "image")
-iface.launch()
-
-
-# gr.Interface(inference, gr.inputs.Image(type="pil"), gr.outputs.Image(type="pil"),title=title,description=description,article=article,examples=examples,allow_flagging=False,allow_screenshot=False).launch(enable_queue=True,cache_examples=True)
+iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+git+https://github.com/huggingface/transformers.git@master#egg=transformers
+torch