jdoexbox360 commited on
Commit
e1bc4b5
1 Parent(s): ac0a21e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,9 +7,9 @@ global tokenizer, model, script_speaker_name, script_responder_name
7
  tokenizer = GPT2Tokenizer.from_pretrained("ethzanalytics/ai-msgbot-gpt2-XL-dialogue")
8
  model = GPT2LMHeadModel.from_pretrained("ethzanalytics/ai-msgbot-gpt2-XL-dialogue", pad_token_id=tokenizer.eos_token_id)
9
 
10
- script_speaker_name = person alpha
11
 
12
- script_responder_name = person beta
13
 
14
  def output(prompt):
15
  sentence = script_speaker_name + ': ' + prompt + '\n' + script_responder_name + ': '
 
7
  tokenizer = GPT2Tokenizer.from_pretrained("ethzanalytics/ai-msgbot-gpt2-XL-dialogue")
8
  model = GPT2LMHeadModel.from_pretrained("ethzanalytics/ai-msgbot-gpt2-XL-dialogue", pad_token_id=tokenizer.eos_token_id)
9
 
10
+ script_speaker_name = "person alpha"
11
 
12
+ script_responder_name = "person beta"
13
 
14
  def output(prompt):
15
  sentence = script_speaker_name + ': ' + prompt + '\n' + script_responder_name + ': '