Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def generate_text(prompt, temperature):
|
|
17 |
|
18 |
print(prompt)
|
19 |
|
20 |
-
prompt_with_eos = "
|
21 |
input_tokens = tokenizer.encode(prompt_with_eos, return_tensors='pt')
|
22 |
|
23 |
input_tokens = input_tokens.to('cpu')
|
|
|
17 |
|
18 |
print(prompt)
|
19 |
|
20 |
+
prompt_with_eos = "#CONTEXT# " + prompt + " #TOPIC# " # Add the string "EOS" to the end of the prompt
|
21 |
input_tokens = tokenizer.encode(prompt_with_eos, return_tensors='pt')
|
22 |
|
23 |
input_tokens = input_tokens.to('cpu')
|