Spaces:
Build error
Build error
dennis-fast
commited on
Commit
•
0640343
1
Parent(s):
5018d35
Update app.py
Browse files
app.py
CHANGED
@@ -4,8 +4,8 @@ import torch
|
|
4 |
|
5 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
6 |
|
7 |
-
tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-
|
8 |
-
model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-
|
9 |
|
10 |
def chat(message, token_response):
|
11 |
token_message = tokenizer.encode(message + tokenizer.eos_token, return_tensors='pt')
|
@@ -17,7 +17,7 @@ input = gr.inputs.Textbox(lines=2, label='User:')
|
|
17 |
output = gr.outputs.Textbox(label='Bot:')
|
18 |
|
19 |
gr.Interface(fn=chat,
|
20 |
-
title="DialoGPT-
|
21 |
inputs=[input, "state"],
|
22 |
outputs=[output, "state"],
|
23 |
allow_screenshot=False,
|
4 |
|
5 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
6 |
|
7 |
+
tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
|
8 |
+
model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
|
9 |
|
10 |
def chat(message, token_response):
|
11 |
token_message = tokenizer.encode(message + tokenizer.eos_token, return_tensors='pt')
|
17 |
output = gr.outputs.Textbox(label='Bot:')
|
18 |
|
19 |
gr.Interface(fn=chat,
|
20 |
+
title="DialoGPT-medium",
|
21 |
inputs=[input, "state"],
|
22 |
outputs=[output, "state"],
|
23 |
allow_screenshot=False,
|