Yash Sachdeva commited on
Commit
be5ca1e
1 Parent(s): 598b840
Files changed (1) hide show
  1. question_paper.py +2 -2
question_paper.py CHANGED
@@ -1,5 +1,5 @@
1
- from transformers import pipeline, set_seed
2
- generator = pipeline('text-generation', model='gpt2')
3
  set_seed(40)
4
 
5
  from fastapi import FastAPI
 
1
+ from transformers import pipeline
2
+ generator = pipeline(model="mrm8488/t5-base-finetuned-question-generation-ap")
3
  set_seed(40)
4
 
5
  from fastapi import FastAPI