update app

app.py

@@ -11,6 +11,7 @@ processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
 cifar100 = CIFAR100(root=os.path.expanduser("~/.cache"), download=True, train=False)
 
 text_inputs = torch.cat([clip.tokenize(f"a photo of a {c}") for c in cifar100.classes])
+text_inputs_2 = ["a photo of a dog", "a photo of a cat"]
 
 
 # TODO debug cette ligne pour avoir un affichage correct
@@ -20,7 +21,7 @@ text_inputs = torch.cat([clip.tokenize(f"a photo of a {c}") for c in cifar100.cl
 
 
 def send_inputs(img):
-    inputs = processor(text=cifar100.classes, images=img, return_tensors="pt", padding=True)
+    inputs = processor(text=text_inputs_2, images=img, return_tensors="pt", padding=True)
     outputs = model(**inputs)
     logits_per_image = outputs.logits_per_image
     probs = logits_per_image.softmax(dim=1)