Spaces:
Runtime error
Runtime error
Maruf commited on
Commit ·
a9284f8
1
Parent(s): e7a1832
cache added
Browse files
app.py
CHANGED
|
@@ -8,6 +8,8 @@ from PIL import Image, UnidentifiedImageError
|
|
| 8 |
import io
|
| 9 |
import os
|
| 10 |
from datetime import datetime
|
|
|
|
|
|
|
| 11 |
|
| 12 |
# Load YOLOv5s pretrained on COCO
|
| 13 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
|
|
|
|
| 8 |
import io
|
| 9 |
import os
|
| 10 |
from datetime import datetime
|
| 11 |
+
import os
|
| 12 |
+
os.environ['TORCH_HOME'] = '/tmp/torch' # Set writable cache for PyTorch Hub
|
| 13 |
|
| 14 |
# Load YOLOv5s pretrained on COCO
|
| 15 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
|