Changes in config

app.py

@@ -15,8 +15,7 @@ HF_API_TOKEN = os.getenv('HF_API_TOKEN')
 ENC_KEY = os.getenv('ENC_KEY')
 dataset_name = os.getenv('DATASET_NAME')
 ds_manager_queue = Queue(maxsize=1)
-# device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-device = torch.device('cpu')
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 def push_files_async():
     try:
@@ -85,4 +84,4 @@ if __name__ == "__main__":
     model_cls.eval()
     feat_extractor.eval()
     demo.queue()
-    demo.launch(share=True)
+    demo.launch()