fix: Updated prediction for better accuracy

app.py

@@ -1,6 +1,7 @@
 import os
 import tensorflow as tf
 import gradio as gr
+import numpy as np
 from huggingface_hub import hf_hub_download
 
 # 1) download your SavedModel from the Hub
@@ -15,8 +16,9 @@ model = tf.keras.models.load_model("./model")
 # 3) simple preprocess + predict
 CLASS_NAMES = ["cat","dog","panda"]
 def predict(image):
-    img = tf.image.resize(image, (64,64)) / 255.0
-    probs = model.predict(img[None,...])[0]
+    resized_image = tf.image.resize(image, (64,64))
+    images_to_predict = np.expand_dims(np.array(resized_image), axis=0)
+    probs = model.predict(images_to_predict)[0]
     return {c: float(p) for c,p in zip(CLASS_NAMES, probs)}
 
 # 4) launch Gradio