Spaces:
Running on Zero
Running on Zero
lixi042 commited on
Commit ·
25eddc4
1
Parent(s): 510e990
Fix: import spaces before CUDA-related packages for HF Spaces compatibility
Browse files
app.py
CHANGED
|
@@ -11,6 +11,12 @@ import tempfile
|
|
| 11 |
from datetime import datetime
|
| 12 |
from pathlib import Path
|
| 13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
# Third-party library imports
|
| 15 |
import cv2
|
| 16 |
import torch
|
|
|
|
| 11 |
from datetime import datetime
|
| 12 |
from pathlib import Path
|
| 13 |
|
| 14 |
+
# HuggingFace Spaces (must be imported before any CUDA-related package)
|
| 15 |
+
try:
|
| 16 |
+
import spaces
|
| 17 |
+
except ImportError:
|
| 18 |
+
pass
|
| 19 |
+
|
| 20 |
# Third-party library imports
|
| 21 |
import cv2
|
| 22 |
import torch
|