TrungNQ commited on
Commit
a663e78
·
verified ·
1 Parent(s): 135593c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -1
app.py CHANGED
@@ -16,6 +16,7 @@ def answer_question(question,context):
16
  return response
17
  grad.Interface(answer_question, inputs=["text","text"], outputs="text").launch()
18
  '''
 
19
  from transformers import pipeline
20
  import gradio as grad
21
  mdl_name = "VietAI/envit5-translation"
@@ -25,4 +26,22 @@ def translate(text):
25
 
26
  response = opus_translator(text)
27
  return response
28
- grad.Interface(translate, inputs=["text",], outputs="text").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  return response
17
  grad.Interface(answer_question, inputs=["text","text"], outputs="text").launch()
18
  '''
19
+ '''
20
  from transformers import pipeline
21
  import gradio as grad
22
  mdl_name = "VietAI/envit5-translation"
 
26
 
27
  response = opus_translator(text)
28
  return response
29
+ grad.Interface(translate, inputs=["text",], outputs="text").launch()
30
+ '''
31
+
32
+ #5.11
33
+
34
+ from transformers import GPT2LMHeadModel,GPT2Tokenizer
35
+ import gradio as grad
36
+
37
+ mdl = GPT2LMHeadModel.from_pretrained('gpt2')
38
+ gpt2_tkn=GPT2Tokenizer.from_pretrained('gpt2')
39
+
40
+ def generate(starting_text):
41
+ tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
42
+ gpt2_tensors = mdl.generate(tkn_ids)
43
+ response = gpt2_tensors
44
+ return response
45
+ txt=grad.Textbox(lines=1, label="English", placeholder="English Text here")
46
+ out=grad.Textbox(lines=1, label="Generated Tensors")
47
+ grad.Interface(generate, inputs=txt, outputs=out).launch()