Update app.py

app.py

@@ -5,13 +5,18 @@ import numpy as np
 from PIL import Image
 
 
-pipe = pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-base-patch16")
+pipes = {
+    "chinese-clip-vit-base-patch16": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-base-patch16"),
+    "chinese-clip-vit-large-patch14": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-large-patch14"),
+    "chinese-clip-vit-large-patch14-336px": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-large-patch14-336px"),
+    "chinese-clip-vit-huge-patch14": pipeline("zero-shot-image-classification", model="OFA-Sys/chinese-clip-vit-huge-patch14"),
+}
 images="festival.jpg"
 
 def shot(image, labels_text):
     PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
     labels = labels_text.split(",")
-    res = pipe(images=PIL_image, 
+    res = pipes['chinese-clip-vit-base-patch16'](images=PIL_image, 
            candidate_labels=labels,
            hypothesis_template= "This is a photo of a {}")
     return {dic["label"]: dic["score"] for dic in res}