RishuD7 commited on
Commit
b6b59b8
1 Parent(s): 4612b54

device changed to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ if __name__ == '__main__':
11
  import gradio as gr
12
  import os
13
 
14
- device = f'cuda' if torch.cuda.is_available() else 'cpu'
15
  config_path = os.path.join('models_file', 'config.pth')
16
  model_path = os.path.join('models_file', 'microsoft-deberta-base_0.9449373420387531_8_best.pth')
17
  tokenizer = AutoTokenizer.from_pretrained('models_file/tokenizer')
 
11
  import gradio as gr
12
  import os
13
 
14
+ device = torch.device('cpu')
15
  config_path = os.path.join('models_file', 'config.pth')
16
  model_path = os.path.join('models_file', 'microsoft-deberta-base_0.9449373420387531_8_best.pth')
17
  tokenizer = AutoTokenizer.from_pretrained('models_file/tokenizer')