smjain commited on
Commit
0feb0a9
1 Parent(s): e2a4d1c

Create new file

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoModelWithLMHead, AutoTokenizer
2
+ import gradio as grad
3
+
4
+ text2text_tkn = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-question-generation-ap")
5
+ mdl = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-question-generation-ap")
6
+
7
+ def text2text(answer,context):
8
+ input_text = "answer: %s context: %s </s>" % (answer, context)
9
+ features = text2text_tkn ([input_text], return_tensors='pt')
10
+
11
+ output = mdl.generate(input_ids=features['input_ids'],
12
+ attention_mask=features['attention_mask'],
13
+ max_length=64)
14
+
15
+ response=text2text_tkn.decode(output[0])
16
+ return response
17
+
18
+ txt=grad.Textbox(lines=10, label="English", placeholder="Context")
19
+ ans=grad.Textbox(lines=1, label="Answer")
20
+ out=grad.Textbox(lines=1, label="Genereated Question")
21
+ grad.Interface(text2text, inputs=[txt,ans], outputs=out).launch()