Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,9 +14,12 @@ import subprocess
|
|
| 14 |
import webbrowser
|
| 15 |
import urllib.parse
|
| 16 |
import http.client
|
|
|
|
| 17 |
import torch
|
| 18 |
import deepspeed # Import DeepSpeed for ZeroGPU
|
| 19 |
|
|
|
|
|
|
|
| 20 |
# Suppress warnings
|
| 21 |
warnings.filterwarnings('ignore', category=UserWarning)
|
| 22 |
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
|
|
|
|
| 14 |
import webbrowser
|
| 15 |
import urllib.parse
|
| 16 |
import http.client
|
| 17 |
+
import spaces
|
| 18 |
import torch
|
| 19 |
import deepspeed # Import DeepSpeed for ZeroGPU
|
| 20 |
|
| 21 |
+
@spaces.GPU
|
| 22 |
+
|
| 23 |
# Suppress warnings
|
| 24 |
warnings.filterwarnings('ignore', category=UserWarning)
|
| 25 |
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
|