Update app.py

app.py

@@ -48,7 +48,8 @@ if file_name is not None:
 
     image = Image.open(file_name)
     col1.image(image, use_column_width=True)
-    pred = catanddog.predict(file_name)[1]
+    image_pred = Image.create(img_bytes)
+    pred = catanddog.predict(image_pred)[1]
     print(pred)
     def metric_rmse(input,target):
         return 100*torch.sqrt(F.mse_loss(F.sigmoid(input.flatten()), target))
@@ -57,11 +58,11 @@ if file_name is not None:
     # learner = learner.load(f'/kaggle/input/puredog/pure-dog-{fold}')
     if pred == 0:
         learn.load('../pure-dog-5')
-        pred = learn.predict(file_name)
+        pred = learn.predict(image_pred)
         print("Score for this doggy:", int(pred[2] * 100))
     if pred == 1:
         learn.load(f'../pure-cat-0')
-        pred = learn.predict(file_name)
+        pred = learn.predict(image_pred)
         print(pred)
         print("Score for this catto:", min(int(pred[2] * 100 + 35), 100))
     col2.header("Probabilities")