Update app.py

app.py

@@ -44,7 +44,7 @@ else:
     for uploaded_file in uploaded_files:
         img = Image.open(uploaded_file)
         st.image(img)
-        input = tf.image_resize(tf.convert_to_tensor(img), [50, 50])
+        input = tf.cast(tf.image.resize(tf.convert_to_tensor(img), [50, 50]), tf.float32) / 255.
         st.write("Model Prediction: " + cnn.apply({"params": params}, input))
         st.write("Model Prediction type: " + type(cnn.apply({"params": params}, input)))
         st.write("Model Prediction type dir: " + dir(cnn.apply({"params": params}, input)))