ttphong68 commited on
Commit
7591ac7
·
1 Parent(s): 9c2abd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -7
app.py CHANGED
@@ -1,10 +1,8 @@
1
  from transformers import GPT2LMHeadModel,GPT2Tokenizer
2
  import gradio as grad
3
-
4
  mdl = GPT2LMHeadModel.from_pretrained('gpt2')
5
  gpt2_tkn=GPT2Tokenizer.from_pretrained('gpt2')
6
 
7
-
8
  def generate(starting_text):
9
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
10
  gpt2_tensors = mdl.generate(tkn_ids,max_length=100,no_repeat_ngram_size=True,num_beams=3,do_sample=True,temperatue=1.5)
@@ -14,11 +12,6 @@ def generate(starting_text):
14
  response=response+f"{i}: {gpt2_tkn.decode(x, skip_special_tokens=True)}"
15
  return response
16
 
17
-
18
-
19
-
20
-
21
-
22
  txt=grad.Textbox(lines=1, label="English", placeholder="English Text here")
23
  out=grad.Textbox(lines=1, label="Generated Tensors")
24
  grad.Interface(generate, inputs=txt, outputs=out).launch()
 
1
  from transformers import GPT2LMHeadModel,GPT2Tokenizer
2
  import gradio as grad
 
3
  mdl = GPT2LMHeadModel.from_pretrained('gpt2')
4
  gpt2_tkn=GPT2Tokenizer.from_pretrained('gpt2')
5
 
 
6
  def generate(starting_text):
7
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
8
  gpt2_tensors = mdl.generate(tkn_ids,max_length=100,no_repeat_ngram_size=True,num_beams=3,do_sample=True,temperatue=1.5)
 
12
  response=response+f"{i}: {gpt2_tkn.decode(x, skip_special_tokens=True)}"
13
  return response
14
 
 
 
 
 
 
15
  txt=grad.Textbox(lines=1, label="English", placeholder="English Text here")
16
  out=grad.Textbox(lines=1, label="Generated Tensors")
17
  grad.Interface(generate, inputs=txt, outputs=out).launch()