Daniel Bustamante Ospina commited on
Commit
608e54d
1 Parent(s): d008b16

Changes in config

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -15,8 +15,7 @@ HF_API_TOKEN = os.getenv('HF_API_TOKEN')
15
  ENC_KEY = os.getenv('ENC_KEY')
16
  dataset_name = os.getenv('DATASET_NAME')
17
  ds_manager_queue = Queue(maxsize=1)
18
- # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
19
- device = torch.device('cpu')
20
 
21
  def push_files_async():
22
  try:
@@ -85,4 +84,4 @@ if __name__ == "__main__":
85
  model_cls.eval()
86
  feat_extractor.eval()
87
  demo.queue()
88
- demo.launch(share=True)
 
15
  ENC_KEY = os.getenv('ENC_KEY')
16
  dataset_name = os.getenv('DATASET_NAME')
17
  ds_manager_queue = Queue(maxsize=1)
18
+ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
19
 
20
  def push_files_async():
21
  try:
 
84
  model_cls.eval()
85
  feat_extractor.eval()
86
  demo.queue()
87
+ demo.launch()