background dict

app.py

@@ -114,8 +114,10 @@ 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_3d/255.0
+        print(img)
+        img_3d=img['background'].reshape(-1,28,28)
+        print(img_3d)
+        img_resized=img/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==3.48.0
+gradio
 
 #streamlit==1.26.0