Update app.py

app.py

@@ -58,7 +58,7 @@ def imshow(img):
     npimg = img.numpy()
     plt.imshow(np.transpose(npimg, (1, 2, 0)))
     plt.axis('off')  # Turn off axis labels
-    st.pyplot()
+    st.pyplot(fig)
 
 # Streamlit app
 st.title("Image Geolocation Prediction")