Hugging Face's logo

Spaces:
Wootang01
/
question_generator_two
Runtime error

App Files Community
Wootang01 commited on
Commit
e678297
•
1 Parent(s): 4662318

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import transformers
8
  from keybert import KeyBERT
9
  from transformers import (T5ForConditionalGeneration, T5Tokenizer)
10
 
11
- DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
12
  MAX_LEN = 512
13
 
14
  tokenizer = T5Tokenizer.from_pretrained('t5-base')
 
8
  from keybert import KeyBERT
9
  from transformers import (T5ForConditionalGeneration, T5Tokenizer)
10
 
11
+ DEVICE = torch.device('cpu')
12
  MAX_LEN = 512
13
 
14
  tokenizer = T5Tokenizer.from_pretrained('t5-base')