Update app.py

app.py

@@ -65,7 +65,8 @@ def create_interface():
             text_features /= text_features.norm(dim=-1, keepdim=True)
 
             text_probs = (100.0 * image_features @ text_features.T).softmax(dim=-1)
-        correct_label = FRUITS30_CLASSES[str(text_probs.argmax().item())]
+        labels1 = labels+[label1,"not a fruit"]
+        correct_label = labels1[(text_probs.argmax().item())]
         return correct_label
 
     with gr.Blocks() as demo: