Update app.py

app.py

@@ -89,11 +89,13 @@ if uploaded_file is not None:
 
     with col2:
         st.button('Detect cavities')      
-        data, wcs = cut(np.log10(data+1), wcs, scale=scale)
+        data, wcs = cut(data, wcs, scale=scale)
 
+        image = np.log10(data+1)
+        
         y_pred = 0
         for j in [0,1,2,3]:
-            rotated = np.rot90(, j)
+            rotated = np.rot90(image, j)
             pred = model.predict(rotated.reshape(1, 128, 128, 1)).reshape(128 ,128)
             pred = np.rot90(pred, -j)
             y_pred += pred / 4