Spaces:

Wootang01
/

question_generator_two

Runtime error

Wootang01 commited on Feb 10, 2022

Commit

61b4399

1 Parent(s): e678297

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ DEVICE = torch.device('cpu')
 MAX_LEN = 512
 tokenizer = T5Tokenizer.from_pretrained('t5-base')
-model = T5ForConditionalGeneration.from_pretrained('Vaibhavbrkn/question-gen')
 mod = KeyBERT('distilbert-base-nli-mean-tokens')
 model.to(DEVICE)

 MAX_LEN = 512
 tokenizer = T5Tokenizer.from_pretrained('t5-base')
+model = T5ForConditionalGeneration.from_pretrained('ZhangCheng/T5-Base-Fine-Tuned-for-Question-Generation')
 mod = KeyBERT('distilbert-base-nli-mean-tokens')
 model.to(DEVICE)