Wootang01 commited on
Commit
ce0ccdb
·
1 Parent(s): 4c3f294

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ num_correct_options = st.number_input('Number of Correction Options', min_value=
9
 
10
  from transformers import T5ForConditionalGeneration, T5Tokenizer
11
  import torch
12
- torch_device = 'cuda' if torch.cude.is_available() else 'cpu'
13
  tokenizer = T5Tokenizer.from_pretrained('deep-learning-analytics/GrammarCorrector')
14
  model = T5ForConditionalGeneration.from_pretrained('deep-learning-analytics/GrammarCorrector').to(torch_device)
15
 
 
9
 
10
  from transformers import T5ForConditionalGeneration, T5Tokenizer
11
  import torch
12
+ torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
13
  tokenizer = T5Tokenizer.from_pretrained('deep-learning-analytics/GrammarCorrector')
14
  model = T5ForConditionalGeneration.from_pretrained('deep-learning-analytics/GrammarCorrector').to(torch_device)
15