isaiah08 commited on
Commit
9727c00
·
1 Parent(s): 94879e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -277,10 +277,10 @@ import gradio as gr
277
  from transformers import pipeline
278
  import tensorflow
279
 
280
- text2text_generator = pipeline("text2text-generation")
281
 
282
  def main(input):
283
- return text2text_generator("question: " + input + " context: " + constitution)
284
 
285
  iface = gr.Interface(fn=main, inputs="text", outputs="text")
286
  iface.launch()
 
277
  from transformers import pipeline
278
  import tensorflow
279
 
280
+ question_answer = pipeline("question-answering")
281
 
282
  def main(input):
283
+ return question_answer(question=input, context=constitution)
284
 
285
  iface = gr.Interface(fn=main, inputs="text", outputs="text")
286
  iface.launch()