Spaces:
Runtime error
Runtime error
Daniel Bustamante Ospina
commited on
Commit
•
608e54d
1
Parent(s):
d008b16
Changes in config
Browse files
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 |
-
|
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(
|
|
|
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()
|