Update app.py

app.py

@@ -69,7 +69,6 @@ def cut(data0, wcs0, scale=1):
 
     return data, wcs
 
-
 # If file is uploaded, read in the data and plot it
 if uploaded_file is not None:
     col1.subheader("Input image")
@@ -84,17 +83,17 @@ if uploaded_file is not None:
         max_scale = int(data.shape[0] // 128)
         # scale = st.slider("Scale", 1, max_scale, 1, 1)
 
-        scale = st.selectbox('Scale:',[i for i in range(max_scale)])
+        scale = int(st.selectbox('Scale:',[i for i in range(max_scale)], label_visibility="hidden"))
 
         plot_image(np.log10(data+1), scale)
 
     with col2:
         st.button('Detect cavities')      
-        data, wcs = cut(data, wcs, scale=scale)
+        data, wcs = cut(np.log10(data+1), wcs, scale=scale)
 
         y_pred = 0
         for j in [0,1,2,3]:
-            rotated = np.rot90(np.log10(data+1), j)
+            rotated = np.rot90(, j)
             pred = model.predict(rotated.reshape(1, 128, 128, 1)).reshape(128 ,128)
             pred = np.rot90(pred, -j)
             y_pred += pred / 4