Spaces:
Runtime error
Runtime error
Commit
·
e1bc4b5
1
Parent(s):
ac0a21e
Update app.py
Browse files
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 + ': '
|