Update app.py

app.py

@@ -36,32 +36,32 @@ def predict(message, history):
     return response.strip()
 
 # 创建Gradio界面
-# demo = gr.ChatInterface(
-#     predict,
-#     title="测试Zhi-writing-dsr1-14b",
-#     description="Zhihu-ai/Zhi-writing-dsr1-14b",
-#     examples=["鲁迅口吻写五百字，描述桔猫的可爱！", "桔了个仔是谁", "介绍自己"],
-#     theme=gr.themes.Soft()
-# )
+demo = gr.ChatInterface(
+    predict,
+    title="测试Zhi-writing-dsr1-14b",
+    description="Zhihu-ai/Zhi-writing-dsr1-14b",
+    examples=["鲁迅口吻写五百字，描述桔猫的可爱！", "桔了个仔是谁", "介绍自己"],
+    theme=gr.themes.Soft()
+)
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# Zhi-writing-dsr1-14")
-    gr.Markdown("这是一个基于Zhi-writing-dsr1-14的文章生成器")
+# with gr.Blocks(theme=gr.themes.Soft()) as demo:
+#     gr.Markdown("# Zhi-writing-dsr1-14")
+#     gr.Markdown("这是一个基于Zhi-writing-dsr1-14的文章生成器")
     
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="输入消息")
-    clear = gr.Button("清除对话")
+#     chatbot = gr.Chatbot()
+#     msg = gr.Textbox(label="输入消息")
+#     clear = gr.Button("清除对话")
     
-    def respond(message, chat_history):
-        bot_message = ""
-        for response in predict(message, chat_history):
-            bot_message = response
-            chat_history.append((message, bot_message))
-            yield chat_history
-        return "", chat_history
+#     def respond(message, chat_history):
+#         bot_message = ""
+#         for response in predict(message, chat_history):
+#             bot_message = response
+#             chat_history.append((message, bot_message))
+#             yield chat_history
+#         return "", chat_history
     
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
+#     msg.submit(respond, [msg, chatbot], [msg, chatbot])
+#     clear.click(lambda: None, None, chatbot, queue=False)
 
 if __name__ == "__main__":
     demo.launch(share=True)