Hugging Face's logo

Spaces:
Wootang01
/
question_generator_two
Runtime error

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ DEVICE = torch.device('cpu')
12
  MAX_LEN = 512
13
 
14
  tokenizer = T5Tokenizer.from_pretrained('t5-base')
15
- model = T5ForConditionalGeneration.from_pretrained('Vaibhavbrkn/question-gen')
16
  mod = KeyBERT('distilbert-base-nli-mean-tokens')
17
  model.to(DEVICE)
18
 
 
12
  MAX_LEN = 512
13
 
14
  tokenizer = T5Tokenizer.from_pretrained('t5-base')
15
+ model = T5ForConditionalGeneration.from_pretrained('ZhangCheng/T5-Base-Fine-Tuned-for-Question-Generation')
16
  mod = KeyBERT('distilbert-base-nli-mean-tokens')
17
  model.to(DEVICE)
18