Fixed errors
Browse files
app.py
CHANGED
@@ -9,7 +9,6 @@ model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
|
|
9 |
model.to(device)
|
10 |
|
11 |
def get_reply(response, history = []):
|
12 |
-
response = input("You: ")
|
13 |
history.append(response)
|
14 |
if response.endswith(("bye", "Bye", "bye.", "Bye.")):
|
15 |
return "<div class='chatbot'>Chatbot restarted</div>", []
|
@@ -19,7 +18,7 @@ def get_reply(response, history = []):
|
|
19 |
outputs = model.generate(**inputs)
|
20 |
reply = tokenizer.decode(outputs[0][1:-1]).strip()
|
21 |
history.append(reply)
|
22 |
-
|
23 |
html = "<div class='chatbot'>"
|
24 |
for m, msg in enumerate(history):
|
25 |
cls = "user" if m%2 == 0 else "bot"
|
@@ -41,5 +40,4 @@ gr.Interface(fn=get_reply,
|
|
41 |
inputs=[gr.inputs.Textbox(placeholder="How are you?"),
|
42 |
"state"],
|
43 |
outputs=["html", "state"],
|
44 |
-
enable_queue=True,
|
45 |
-
css=css).launch(debug=True)
|
|
|
9 |
model.to(device)
|
10 |
|
11 |
def get_reply(response, history = []):
|
|
|
12 |
history.append(response)
|
13 |
if response.endswith(("bye", "Bye", "bye.", "Bye.")):
|
14 |
return "<div class='chatbot'>Chatbot restarted</div>", []
|
|
|
18 |
outputs = model.generate(**inputs)
|
19 |
reply = tokenizer.decode(outputs[0][1:-1]).strip()
|
20 |
history.append(reply)
|
21 |
+
|
22 |
html = "<div class='chatbot'>"
|
23 |
for m, msg in enumerate(history):
|
24 |
cls = "user" if m%2 == 0 else "bot"
|
|
|
40 |
inputs=[gr.inputs.Textbox(placeholder="How are you?"),
|
41 |
"state"],
|
42 |
outputs=["html", "state"],
|
43 |
+
css=css).launch(debug=True, enable_queue=True, share=True)
|
|