Update app.py

app.py

@@ -34,11 +34,10 @@ def predict(img) -> Tuple[Dict, float]:
     """Transforms and performs a prediction on img and returns prediction and time taken."""
     start_time = timer()
 
-    # ✅ Convert numpy image (dtype=uint8) to PIL.Image (RGB)
     if isinstance(img, np.ndarray):
-        img = Image.fromarray(img.astype("uint8"), mode="RGB")
+        img = img.astype(np.uint8)  # ✅ Ensure dtype is uint8
+        img = Image.fromarray(img, mode="RGB")  # ✅ Safe conversion
 
-    # Now it's safe to use vit_transforms
     img = vit_transforms(img).unsqueeze(0)
 
     vit.eval()
@@ -52,6 +51,7 @@ def predict(img) -> Tuple[Dict, float]:
     pred_time = round(timer() - start_time, 5)
     return pred_labels_and_probs, pred_time
 
+
 ### 4. Gradio app ###
 
 title = "VisionBite 🍔👁"