comrender commited on
Commit
4a135c6
·
verified ·
1 Parent(s): b4f8216

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -35,12 +35,8 @@ css = """
35
  """
36
 
37
  # Device setup
38
- if torch.cuda.is_available():
39
- power_device = "GPU"
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")