Update app.py

app.py

@@ -17,7 +17,7 @@ def predict_image(img):
   img = transforms.ToPILImage()(img)
   img = transforms.Resize((256,256))(img)
   prediction=clf.predict(img)
-  print(clf(image))
+  print(prediction)
   print(model.config.id2label[predicted_label])
   return {class_names[i]: float(prediction[i]["score"]) for i in range(2)}