forgot to add files

pages/2_Perform_Crop_Classification.py

@@ -5,10 +5,12 @@ from folium import Icon
 
 from messis.messis import Messis
 from inference import perform_inference
+from inference import generate_presigned_url
 
-st.set_page_config(layout="wide")
+from dotenv import load_dotenv
 
-GEOTIFF_PATH = "./data/stacked_features.tif"
+load_dotenv()
+st.set_page_config(layout="wide")
 
 # Load the model
 @st.cache_resource
@@ -78,19 +80,10 @@ def perform_inference_step():
         )
         st.success("Inference completed!")
 
-    # GeoTIFF Satellite Imagery with selected timestep and band
-    # m.add_raster(
-    #     GEOTIFF_PATH,
-    #     layer_name="Satellite Image",
-    #     bands=selected_bands,
-    #     fit_bounds=True,
-    #     vmin=vmin_vmax[selected_band][0],
-    #     vmax=vmin_vmax[selected_band][1],
-    # )
-
     # Add COG
+    presigned_url = generate_presigned_url('messis-demo', 'stacked_features_cog.tif')
     m.add_cog_layer(
-        url="https://messis-demo.s3.amazonaws.com/stacked_features_cog.tif",
+        url=presigned_url,
         name="Sentinel-2 Satellite Imagery",
         bands=selected_bands,
         rescale=f"{vmin_vmax[selected_band][0]},{vmin_vmax[selected_band][1]}",

requirements.txt

@@ -20,4 +20,5 @@ numpy==1.26.4
 lion-pytorch==0.2.2
 timm==0.9.16
 pyproj
-boto3
+boto3
+python-dotenv