run480 commited on
Commit
4a9caee
·
verified ·
1 Parent(s): 9e5afce

Update app.py

Browse files

Decode tensors into text

Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -172,7 +172,10 @@ gpt2_tkn=GPT2Tokenizer.from_pretrained('gpt2')
172
  def generate(starting_text):
173
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
174
  gpt2_tensors = mdl.generate(tkn_ids)
175
- response = gpt2_tensors
 
 
 
176
  return response
177
 
178
  txt=grad.Textbox(lines=1, label="English", placeholder="English Text here")
 
172
  def generate(starting_text):
173
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
174
  gpt2_tensors = mdl.generate(tkn_ids)
175
+ response=""
176
+     #response = gpt2_tensors
177
+     for i, x in enumerate(gpt2_tensors):
178
+        response=response+f"{i}: {gpt2_tkn.decode(x, skip_special_tokens=True)}" # Decode tensors into text
179
  return response
180
 
181
  txt=grad.Textbox(lines=1, label="English", placeholder="English Text here")