macadeliccc commited on
Commit
f4995ab
Β·
1 Parent(s): 2593c1c
Files changed (1) hide show
  1. app.py +4 -8
app.py CHANGED
@@ -4,16 +4,12 @@ import torch
4
  import subprocess
5
  import numpy as np
6
 
7
- device_available = print(torch.cuda.is_available()) # <-- True πŸ€—
8
- device_count = print(torch.cuda.device_count()) # <-- 1 πŸ€—
9
- device_name = print(torch.cuda.get_device_name()) # <-- 'A10G' πŸ€—
 
10
  @spaces.GPU
11
  def start_ochat_server():
12
- global device_available, device_count, device_name
13
-
14
- print(device_available)
15
- print(device_name) # <-- 'cuda:0' πŸ€—
16
- print(device_count) # <-- 1 πŸ€—
17
 
18
  # Command to start the ochat inference server
19
  command = [
 
4
  import subprocess
5
  import numpy as np
6
 
7
+ print(torch.cuda.is_available()) # <-- True πŸ€—
8
+ print(torch.cuda.device_count()) # <-- 1 πŸ€—
9
+ print(torch.cuda.get_device_name()) # <-- 'A10G' πŸ€—
10
+
11
  @spaces.GPU
12
  def start_ochat_server():
 
 
 
 
 
13
 
14
  # Command to start the ochat inference server
15
  command = [