Update app.py

app.py

@@ -5,7 +5,7 @@ from off_topic import OffTopicDetector
 detector = OffTopicDetector("openai/clip-vit-base-patch32")
 
 def validate(item_id: str, threshold: float):
-    images, domain, probas, valid_probas, invalid_probas = detector.predict_item_probas(item_id)
+    images, domain, probas, valid_probas, invalid_probas = detector.predict_probas_item(item_id)
     valid_images = [x for i, x in enumerate(images) if valid_probas[i].squeeze() >= threshold]
     invalid_images = [x for i, x in enumerate(images) if valid_probas[i].squeeze() < threshold]
     return f"<h2>Domain: {domain}</h2>", valid_images, invalid_images