Add changes

app.py

@@ -5,6 +5,8 @@ import torch
 from model import create_effnetb4_model
 from timeit import default_timer as timer
 from typing import Tuple, Dict
+import numpy as np
+from PIL import Image
 
 class_names = ['apple_pie','baby_back_ribs','baklava','beef_carpaccio','beef_tartare','beet_salad','beignets','bibimbap','bread_pudding',
  'breakfast_burrito','bruschetta','caesar_salad','cannoli','caprese_salad','carrot_cake','ceviche','cheese_plate','cheesecake','chicken_curry',
@@ -40,6 +42,10 @@ effnetb4.load_state_dict(state_dict)
 def predict(img) -> Tuple[Dict, float]:
     """Transforms and performs a prediction on img and returns prediction and time taken."""
     start_time = timer()
+
+    if isinstance(img, np.ndarray):
+        img = Image.fromarray(img)
+
     img = effnetb4_transforms(img).unsqueeze(0)
 
     effnetb4.eval()

requirements.txt

@@ -1,3 +1,5 @@
 torch==2.2.0
 torchvision==0.17.0
 gradio==4.44.0
+Pillow==8.4.0
+numpy<2