sohailq commited on
Commit
93d45c5
1 Parent(s): 9514abc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -9
app.py CHANGED
@@ -1,11 +1,20 @@
1
- from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 
 
 
 
 
 
 
 
 
 
 
 
2
  import gradio as grad
3
- import ast
4
- mdl_name = "deepset/roberta-base-squad2"
5
- my_pipeline = pipeline('question-answering', model=mdl_name,tokenizer=mdl_name)
6
- def answer_question(question,context):
7
- text= "{"+"'question': '"+question+"','context':'"+context+"'}"
8
- di=ast.literal_eval(text)
9
- response = my_pipeline(di)
10
  return response
11
- grad.Interface(answer_question, inputs=["text","text"],outputs="text").launch()
 
1
+ #from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
2
+ #import gradio as grad
3
+ #import ast
4
+ #mdl_name = "deepset/roberta-base-squad2"
5
+ #my_pipeline = pipeline('question-answering', model=mdl_name,tokenizer=mdl_name)
6
+ #def answer_question(question,context):
7
+ #text= "{"+"'question': '"+question+"','context':'"+context+"'}"
8
+ #di=ast.literal_eval(text)
9
+ #response = my_pipeline(di)
10
+ #return response
11
+ #grad.Interface(answer_question, inputs=["text","text"],outputs="text").launch()
12
+
13
+ from transformers import pipeline
14
  import gradio as grad
15
+ mdl_name = "Helsinki-NLP/opus-mt-en-de"
16
+ opus_translator = pipeline("translation", model=mdl_name)
17
+ def translate(text):
18
+ response = opus_translator(text)
 
 
 
19
  return response
20
+ grad.Interface(translate, inputs=["text",], outputs="text").launch()