SinaRp commited on
Commit
cf48ce1
·
verified ·
1 Parent(s): c5f0901

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  from transformers import pipeline
4
 
5
  def load_question_generator(num_questions):
6
- pipe = pipeline("text2text-generation", model="SinaRp/Question_generator_persian", do_sample=True, top_k=50, top_p=0.95, num_return_sequences=num_questions)
7
  return pipe
8
 
9
  def generate_questions(context, num_questions):
 
3
  from transformers import pipeline
4
 
5
  def load_question_generator(num_questions):
6
+ pipe = pipeline("text2text-generation", model="SinaRp/Question_generator_persian", do_sample=True, top_k=5, num_return_sequences=num_questions)
7
  return pipe
8
 
9
  def generate_questions(context, num_questions):