salomonsky commited on
Commit
6459222
1 Parent(s): 4b73d5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,8 +11,9 @@ tokenizer = GPT2Tokenizer.from_pretrained('checkpoints')
11
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
12
 
13
  def generate_output(name, date_of_birth):
 
14
  welcome = f"Bienvenido {name}:"
15
- input_tokens = tokenizer.encode(add_special_tokens=True)
16
  input_text = tokenizer.decode(input_tokens)
17
  gpt2_output = welcome + generator(input_text, max_length=60, do_sample=True, temperature=0.6)
18
  generated_text = gpt2_output[0]['generated_text']
 
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']