Update app.py

app.py

@@ -25,7 +25,7 @@ model = GPT4All(model_name=model_name,
                 device="cpu")
 print("Finish the model init process")
 
-def generater(message, history):
+def generater(message):
     prompt = message + prompt_suffix
     max_new_tokens = round(len(prompt) / 3.0 * 1.2)
     outputs = []
@@ -43,24 +43,17 @@ def generater(message, history):
     transferred_completion = utils.transfer_llm_completion(completion, message)
     yield transferred_completion
 
-print("Create chatbot")
-chatbot = gr.Chatbot()
-print("Created chatbot")
 
-iface = gr.ChatInterface(
+demo = gr.Interface(
     fn = generater,
     title=title,
     description = description,
-    chatbot=chatbot,
-    additional_inputs=[],
+    inputs=["text"],
+    outputs=["text"],
     examples=[
         ["<speaker:1> Hello, my name is Tom. May I speak to Laura <speaker:2> please? Hello, this is Laura. <speaker:1> Hi Laura, how are you? This is <speaker:2> Tom. Hi Tom, I haven't seen you for a <speaker:1> while."],
    ]
 )
 
-with gr.Blocks() as demo:
-    iface.render()
-
-
 if __name__ == "__main__":
     demo.queue(max_size=3).launch()