salomonsky commited on
Commit
0aa161b
·
1 Parent(s): 6459222

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -11,11 +11,10 @@ tokenizer = GPT2Tokenizer.from_pretrained('checkpoints')
11
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
12
 
13
  def generate_output(name, date_of_birth):
14
- prompt = f""
15
- welcome = f"Bienvenido {name}:"
16
  input_tokens = tokenizer.encode(prompt, add_special_tokens=True)
17
  input_text = tokenizer.decode(input_tokens)
18
- gpt2_output = welcome + generator(input_text, max_length=60, do_sample=True, temperature=0.6)
19
  generated_text = gpt2_output[0]['generated_text']
20
  generated_text = generated_text.replace(input_text, "").strip()
21
  if len(gpt2_output) == 0 or 'generated_text' not in gpt2_output[0]:
 
11
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
12
 
13
  def generate_output(name, date_of_birth):
14
+ prompt = f"Bienvenido {name}:"
 
15
  input_tokens = tokenizer.encode(prompt, add_special_tokens=True)
16
  input_text = tokenizer.decode(input_tokens)
17
+ gpt2_output = generator(input_text, max_length=60, do_sample=True, temperature=0.6)
18
  generated_text = gpt2_output[0]['generated_text']
19
  generated_text = generated_text.replace(input_text, "").strip()
20
  if len(gpt2_output) == 0 or 'generated_text' not in gpt2_output[0]: