update argument

app.py

@@ -20,8 +20,8 @@ st.write("""
 file = st.file_uploader("Upload an X-ray image", type= ['png', 'jpg'])
 
 
-def model_prediction(img, model):
-    resize = tf.image.resize(img, (256,256))
+def model_prediction(image, model):
+    resize = tf.image.resize(image, (256,256))
     yhat = model.predict(np.expand_dims(resize/255, 0))
     if(yhat>0.5):
         result = "Prediction is loose"
@@ -34,7 +34,7 @@ if file is None:
 else:
     image = Image.open(file)
     st.image(image, use_column_width=True)
-    predictions = model_prediction(image, model)
+    predictions = model_prediction(file, model)
     st.write(prediction)
     print(
     "This image most likely belongs to {}."