Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,8 +1,14 @@
+import os
 import streamlit as st
 import torch
 import numpy as np
 from streamlit_webrtc import webrtc_streamer, VideoTransformerBase
 
+# Set a writable cache directory for PyTorch
+torch_cache_dir = os.path.join(os.getcwd(), 'torch_cache')
+os.makedirs(torch_cache_dir, exist_ok=True)
+os.environ['TORCH_HOME'] = torch_cache_dir
+
 # Load the YOLOv5 model
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='model/best.pt', force_reload=True)