gradio 3.48.0

app.py

@@ -114,8 +114,8 @@ def predict_digit(img):
         #img_data = im.fromarray(img)
         #img_data.save(f"image1.jpg")
         #count=count+1
-        # img_3d=img.reshape(-1,28,28)
-        img_resized=img/255.0
+        img_3d=img.reshape(-1,28,28)
+        img_resized=img_3d/255.0
         pred_prob=loaded_model.predict(img_resized)
 
         pred_prob=pred_prob*100

requirements.txt

@@ -5,6 +5,6 @@ scikit_learn<1.3.0
 seaborn==0.12.2
 
 #torchvision<0.15.2
-gradio
+gradio==3.48.0
 
 #streamlit==1.26.0