lixi042 commited on
Commit
25eddc4
·
1 Parent(s): 510e990

Fix: import spaces before CUDA-related packages for HF Spaces compatibility

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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