Spaces:
Sleeping
Sleeping
Commit
路
6459222
1
Parent(s):
4b73d5c
Update app.py
Browse files
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']
|