anezatra commited on
Commit
9a3ea39
·
verified ·
1 Parent(s): b7d11fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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, history):
9
- prompt = "\n".join(history + [message])
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, history):
46
- response = respond(user_message, history)
47
- return "", history + [[user_message, response]]
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)