Update app.py

app.py

@@ -40,25 +40,16 @@ def classify_food_vs_nonfood(image):
         # Create a draw object
         draw = ImageDraw.Draw(image)
         
-        # Specify font and size
-        font = ImageFont.load_default()
-        
-        # Get text size
-        text_font = ImageFont.truetype("Hack-Regular.ttf", 24)
-        text_bbox = draw.textbbox((0, 0), label, font=text_font)
-        text_size = (text_bbox[2] - text_bbox[0], text_bbox[3] - text_bbox[1])
-        
-        # Calculate text position
-        text_position = ((image_size[0] - text_size[0]) // 2, 10)
-        
         # Determine label color and font size based on prediction
         if final_prediction == 0:
             label_color = (255, 0, 0)  # Red for "Food"
             text_font = ImageFont.truetype("Hack-Regular.ttf", 24)  # Font size for "Food"
+            text_bbox = draw.textbbox((0, 0), label, font=text_font)
+            text_size = (text_bbox[2] - text_bbox[0], text_bbox[3] - text_bbox[1])
+            text_position = ((image_size[0] - text_size[0]) // 2, 10)
         else:
             label_color = (0, 255, 0)  # Green for "Non Food"
-            text_font = ImageFont.truetype("Hack-Regular.ttf", 36)  # Font size for "Non Food"
-            # Calculate text position for larger font size
+            text_font = ImageFont.truetype("Hack-Regular.ttf", 48)  # Increased font size for "Non Food"
             text_bbox = draw.textbbox((0, 0), label, font=text_font)
             text_size = (text_bbox[2] - text_bbox[0], text_bbox[3] - text_bbox[1])
             text_position = ((image_size[0] - text_size[0]) // 2, (image_size[1] - text_size[1]) // 2)