Update app.py

app.py

@@ -1,3 +1,13 @@
+# https://discuss.huggingface.co/t/how-to-fix-workload-evicted-storage-limit-exceeded-50g-error-in-huggingface-spaces/169258
+import os
+import torch
+
+# CRITICAL: Redirect cache to temporary storage
+os.environ['TORCH_HOME'] = '/tmp/torch_cache'
+os.environ['HUB_DIR'] = '/tmp/torch_hub'
+os.environ['TMPDIR'] = '/tmp'
+torch.hub.set_dir('/tmp/torch_hub')
+
 import glob
 import gradio as gr
 from ultralytics import YOLO