Spaces:
Running
Running
Commit
·
0aa161b
1
Parent(s):
6459222
Update app.py
Browse files
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 =
|
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]:
|