Update app.py

app.py

@@ -78,7 +78,6 @@ if uploaded_file is not None:
     with fits.open(uploaded_file) as hdul:
         data = hdul[0].data
         wcs = WCS(hdul[0].header)
-        image = np.log10(data+1)
 
     # Add a slider to change the scale
     with col1:
@@ -87,7 +86,7 @@ if uploaded_file is not None:
 
         scale = st.selectbox('Scale:',[i for i in range(max_scale)])
 
-        plot_image(image, scale)
+        plot_image(np.log10(data+1), scale)
 
     with col2:
         st.button('Detect cavities')      
@@ -95,7 +94,7 @@ if uploaded_file is not None:
 
         y_pred = 0
         for j in [0,1,2,3]:
-            rotated = np.rot90(image, j)
+            rotated = np.rot90(np.log10(data+1), j)
             pred = model.predict(rotated.reshape(1, 128, 128, 1)).reshape(128 ,128)
             pred = np.rot90(pred, -j)
             y_pred += pred / 4