imabackstabber commited on
Commit
a60a640
·
1 Parent(s): 669c2e0

test mmdet pipeline

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -21,12 +21,13 @@ os.makedirs(OUT_FOLDER, exist_ok=True)
21
  @spaces.GPU(enable_queue=True)
22
  def infer(image_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
23
  num_gpus = 1 if torch.cuda.is_available() else -1
24
- print("!!! torch.cuda.is_available: ", torch.cuda.is_available())
25
- print("!!! torch.cuda.device_count: ", torch.cuda.device_count())
26
- print("CUDA version: ", torch.version.cuda)
27
- index = torch.cuda.current_device()
28
- print("CUDA current_device: ", index)
29
- print("CUDA device_name: ", torch.cuda.get_device_name(index))
 
30
  from main.inference import Inferer
31
  inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
32
  os.system(f'rm -rf {OUT_FOLDER}/*')
 
21
  @spaces.GPU(enable_queue=True)
22
  def infer(image_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
23
  num_gpus = 1 if torch.cuda.is_available() else -1
24
+ # dismiss cuda information
25
+ # print("!!! torch.cuda.is_available: ", torch.cuda.is_available())
26
+ # print("!!! torch.cuda.device_count: ", torch.cuda.device_count())
27
+ # print("CUDA version: ", torch.version.cuda)
28
+ # index = torch.cuda.current_device()
29
+ # print("CUDA current_device: ", index)
30
+ # print("CUDA device_name: ", torch.cuda.get_device_name(index))
31
  from main.inference import Inferer
32
  inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
33
  os.system(f'rm -rf {OUT_FOLDER}/*')