Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -35,12 +35,8 @@ css = """
|
|
35 |
"""
|
36 |
|
37 |
# Device setup
|
38 |
-
|
39 |
-
|
40 |
-
device = "cuda"
|
41 |
-
else:
|
42 |
-
power_device = "CPU"
|
43 |
-
device = "cpu"
|
44 |
|
45 |
# Get HuggingFace token
|
46 |
huggingface_token = os.getenv("HF_TOKEN")
|
|
|
35 |
"""
|
36 |
|
37 |
# Device setup
|
38 |
+
power_device = "ZeroGPU"
|
39 |
+
device = "cpu"
|
|
|
|
|
|
|
|
|
40 |
|
41 |
# Get HuggingFace token
|
42 |
huggingface_token = os.getenv("HF_TOKEN")
|