FrederikRautenberg commited on
Commit
091b3ba
·
1 Parent(s): b198353

debug threads

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -12,6 +12,7 @@ import gradio as gr
12
  from pvq_manipulation.helper.creapy_wrapper import process_file
13
  from creapy.utils import config
14
 
 
15
  torch.set_num_threads(os.cpu_count() or 1)
16
 
17
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
12
  from pvq_manipulation.helper.creapy_wrapper import process_file
13
  from creapy.utils import config
14
 
15
+ import os
16
  torch.set_num_threads(os.cpu_count() or 1)
17
 
18
  device = 'cuda' if torch.cuda.is_available() else 'cpu'