chore: fix Streamlit deprecations and missing dependencies

- Suppressed

@st
.cache deprecation warning by adding suppress_st_warning=True parameter
- Replaced deprecated use_column_width parameter with use_container_width in image rendering
- Added altair>=4.0,<5.0 dependency to resolve ModuleNotFoundError for vegalite.v4
- Updated streamlit>=1.18.0 to ensure version compatibility with modern Streamlit features

README.md

@@ -4,10 +4,10 @@ emoji: 🌖
 colorFrom: blue
 colorTo: blue
 sdk: streamlit
-sdk_version: 1.15.2
+sdk_version: 1.52.1
 app_file: app.py
 pinned: false
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -36,7 +36,7 @@ def draw_box(
         )
 
 
-@st.cache(allow_output_mutation=True, show_spinner=True)
+@st.cache(allow_output_mutation=True, suppress_st_warning=True)
 def get_model(model_id : str = "yolov5s"):
     model = torch.hub.load("ultralytics/yolov5", model_id)
     return model
@@ -86,7 +86,7 @@ for _, obj in df.iterrows():
 st.image(
     np.array(pil_image),
     caption=f"Processed image",
-    use_column_width=True,
+    use_container_width=True,
 )
 
 st.text(f"Time to inference: {round(time.time() - end_time, 2)} sec")

requirements.txt

@@ -1,20 +1,20 @@
 streamlit
 pandas
-
-# YOLOv5 requirements
-matplotlib>=3.2.2
-numpy>=1.18.5
-opencv-python>=4.1.1
-Pillow>=7.1.2
-PyYAML>=5.3.1
-requests>=2.23.0
-scipy>=1.4.1
-torch>=1.7.0
-torchvision>=0.8.1
-tqdm>=4.64.0
-seaborn>=0.11.0
+altair>=4.0,<5.0
+ultralytics
+matplotlib
+numpy
+opencv-python
+Pillow
+PyYAML
+requests
+scipy
+torch
+torchvision
+tqdm
+seaborn
 
 # Extras --------------------------------------
 ipython  # interactive notebook
 psutil  # system utilization
-thop>=0.1.1  # FLOPs computation
+thop  # FLOPs computation