try except

food_nonfood.py

@@ -21,10 +21,16 @@ classifier = pipeline("zero-shot-classification", model="roberta-large-mnli")
 
 # Classify item as food or non-food
 def classify_as_food_nonfood(item):
-    cleaned_item = item.strip().lower()
-    result = classifier(cleaned_item, candidate_labels=["food", "non-food"])
-    label = result["labels"][0]
-    score = result["scores"][0]
+    try:
+        cleaned_item = item.strip().lower()
+        result = classifier(cleaned_item, candidate_labels=["food", "non-food"])
+        label = result["labels"][0]
+        score = result["scores"][0]
+    except Exception as e:
+        print(f"Error: {e}")
+        print(f"item is: {item}")
+        label = "non-food"
+        score = 0.0
     # print(f"Item: {item}, Label: {label}, Score: {score}")
     return label, score