Akbartus commited on
Commit
c674488
·
1 Parent(s): 3da047b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,7 +4,10 @@ from transformers import AutoModelWithLMHead, AutoTokenizer
4
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-question-generation-ap")
5
  model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-question-generation-ap")
6
 
7
- def get_questions(input_data, max_length=64): context = input_data[0] answers = input_data[1:] generated_questions = []
 
 
 
8
 
9
  for answer in answers:
10
  input_text = "answer: %s context: %s </s>" % (answer, context)
 
4
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-question-generation-ap")
5
  model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-question-generation-ap")
6
 
7
+ def get_questions(input_data, max_length=64):
8
+ context = input_data[0]
9
+ answers = input_data[1:]
10
+ generated_questions = []
11
 
12
  for answer in answers:
13
  input_text = "answer: %s context: %s </s>" % (answer, context)