Update app.py

app.py

@@ -345,7 +345,7 @@ if st.button("Run Inference"):
             
             # After getting prediction from model
             pred_label_mask = np.argmax(pred_masks[0], axis=-1)  # shape: (H, W)
-            st.success(f "Unique classes in mask: {np.unique(pred_label_mask)}")
+            st.success(f" Unique classes in mask: {np.unique(pred_label_mask)}")
             
             # Get georeference transform from TIF image
             transform, crs = get_transform_from_tif(optic_path)