skylersterling commited on
Commit
a6faa42
·
verified ·
1 Parent(s): a819a8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def generate_text(prompt, temperature):
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')
 
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')