Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers import pipeline
 
 asr = pipeline("automatic-speech-recognition")
 latent = gr.Interface.load("spaces/multimodalart/latentdiffusion")
-zero = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch32")
+zero = pipeline("zero-shot-image-classification")
 #zero = gr.Interface.load("spaces/Datatrooper/zero-shot-image-classification")
 #tts = gr.Interface.load("spaces/osanseviero/tortoisse-tts")
 
@@ -49,9 +49,9 @@ def zero_shot(image, text_input):
     return {dic["label"]: dic["score"] for dic in res}
 
 def shot(image, labels_text):
-    #PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
+    PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
     labels = labels_text.split(",")
-    res = pipe(images= image, 
+    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}