Update app.py

app.py

@@ -103,22 +103,33 @@ except Exception as e:
 
 def classify_food(image):
     """Classify food image using the pre-trained model"""
+    print("classify_food function called") # Check if this function is even called
     try:
-        if model_loaded:
-            if isinstance(image, np.ndarray):
-                image = Image.fromarray(image)
-            image = processor(images=image, return_tensors="pt")
-            with torch.no_grad():
-                outputs = model(**image)
-            predicted_idx = torch.argmax(outputs.logits, dim=-1).item()
-            food_name = model.config.id2label.get(predicted_idx, "Unknown Food")
-            return food_name.lower() #Convert classification to lowercase
-        else:
+        if not model_loaded:
+            print("Model not loaded, returning 'Unknown'")
             return "Unknown"
 
+        print(f"Image type: {type(image)}") # Check the type of the image
+
+        if isinstance(image, np.ndarray):
+            print("Image is a numpy array, converting to PIL Image")
+            image = Image.fromarray(image)
+
+        print(f"Image mode: {image.mode}") # Check image mode (e.g., RGB, L)
+
+        image = processor(images=image, return_tensors="pt")
+        print(f"Processed image: {image}")  # Print the output of the processor
+
+        with torch.no_grad():
+            outputs = model(**image)
+        predicted_idx = torch.argmax(outputs.logits, dim=-1).item()
+        food_name = model.config.id2label.get(predicted_idx, "Unknown Food")
+        print(f"Predicted food name: {food_name}") # Print the predicted food name
+        return food_name.lower()  # Convert classification to lowercase
+
     except Exception as e:
-        print("Classify food error", str(e))
-        return "Unknown" #If a exception arises make sure to create a default case
+        print(f"Classify food error: {e}")  # Print the full error message
+        return "Unknown"  # If an exception arises make sure to create a default case
 
 # -------------------------------------------------
 # USDA API Integration - REMOVED for local HF Spaces deployment