Update app.py

app.py

@@ -1,4 +1,3 @@
-'''
 import transformers
 import gradio as gr
 import torch
@@ -28,8 +27,8 @@ def predict(input, history=[]):
                             
     # convert the tokens to text, and then split the responses into lines
     response = tokenizer.decode(history[0]).split("<|endoftext|>")
-    #response = [(response[i], response[i+1]) for i in range(0, len(response)-1, 2)]  # convert to tuples of list
-    response.remove("")
+    response = [(response[i], response[i+1]) for i in range(0, len(response)-1, 2)]  # convert to tuples of list
+    #response.remove("")
     
     # write some HTML
     html = "<div class='chatbot'>"
@@ -38,8 +37,8 @@ def predict(input, history=[]):
         html += "<div class='msg {}'> {}</div>".format(cls, msg)
     html += "</div>"
     
-    #return response, history
-    return html, history
+    return response, history
+    #return html, history
     
 css = """
 .chatbox {display:flex;flex-direction:column}
@@ -50,14 +49,14 @@ css = """
 """
 
 gr.Interface(fn=predict,
-    theme="default",
+    theme="grass",
     title="DialoGPT-large",
     inputs=[gr.inputs.Textbox(placeholder="How are you?"), "state"],
     outputs=["html", "state"],
-    css=css
+    #css=css
     ).launch()
-'''
 
+'''
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
@@ -102,4 +101,5 @@ gr.Interface(fn=predict,
              inputs=[gr.inputs.Textbox(placeholder="How are you?"), "state"],
              outputs=["html", "state"],
              css=css).launch()
+'''