Spaces:

xelpmocAI
/

entity-extraction

Runtime error

RishuD7 commited on Jun 9, 2023

Commit

b6b59b8

1 Parent(s): 4612b54

device changed to cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ if __name__ == '__main__':
     import gradio as gr
     import os
-    device = f'cuda' if torch.cuda.is_available() else 'cpu'
     config_path = os.path.join('models_file', 'config.pth')
     model_path  = os.path.join('models_file', 'microsoft-deberta-base_0.9449373420387531_8_best.pth')
     tokenizer = AutoTokenizer.from_pretrained('models_file/tokenizer')

     import gradio as gr
     import os
+    device = torch.device('cpu')
     config_path = os.path.join('models_file', 'config.pth')
     model_path  = os.path.join('models_file', 'microsoft-deberta-base_0.9449373420387531_8_best.pth')
     tokenizer = AutoTokenizer.from_pretrained('models_file/tokenizer')