Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ model_name = "anezatra/gpt2_openassistant_guanaco"
|
|
5 |
model = GPT2LMHeadModel.from_pretrained(model_name)
|
6 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
7 |
|
8 |
-
def respond(message
|
9 |
-
prompt = "
|
10 |
|
11 |
inputs = tokenizer.encode(prompt, return_tensors="pt")
|
12 |
outputs = model.generate(
|
@@ -42,9 +42,9 @@ with gr.Blocks(theme=gr.Theme.from_hub('gradio/monochrome')) as demo:
|
|
42 |
clear = gr.Button("Clear chat")
|
43 |
submit = gr.Button("Send message")
|
44 |
|
45 |
-
def user_input(user_message
|
46 |
-
response = respond(user_message
|
47 |
-
return "",
|
48 |
|
49 |
msg.submit(user_input, [msg, chatbot], [msg, chatbot], queue=False)
|
50 |
clear.click(lambda: None, None, chatbot, queue=False)
|
|
|
5 |
model = GPT2LMHeadModel.from_pretrained(model_name)
|
6 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
7 |
|
8 |
+
def respond(message):
|
9 |
+
prompt = f"{message}"
|
10 |
|
11 |
inputs = tokenizer.encode(prompt, return_tensors="pt")
|
12 |
outputs = model.generate(
|
|
|
42 |
clear = gr.Button("Clear chat")
|
43 |
submit = gr.Button("Send message")
|
44 |
|
45 |
+
def user_input(user_message):
|
46 |
+
response = respond(user_message)
|
47 |
+
return "", response
|
48 |
|
49 |
msg.submit(user_input, [msg, chatbot], [msg, chatbot], queue=False)
|
50 |
clear.click(lambda: None, None, chatbot, queue=False)
|