Fix docs

.ruff.toml

@@ -0,0 +1,21 @@
+# Assume Python 3.10.
+target-version = "py310"
+# Decrease the maximum line length to 79 characters.
+line-length = 300
+
+# pyflakes, pycodestyle, isort
+# flake8 YTT, pydocstyle D, pylint PLC
+select = ["F", "E", "W", "I001", "YTT", "D", "PLC"]
+# select = ["ALL"]
+
+# E501 Line too long
+# D102 Missing docstring in public method
+# D100 Missing docstring in public module
+# E501 Line too long
+# D103 Missing docstring in public function
+# D101 Missing docstring in public class
+# `multi-line-summary-first-line` (D212)
+# `one-blank-line-before-class` (D203)
+extend-ignore = ["E501", "D100", "D101", "D102", "D103", "D212", "D203"]
+
+exclude = [".venv", "ultrachat-13B-test.py"]

app.py

@@ -1,6 +1,4 @@
-"""Credit to https://github.com/THUDM/ChatGLM2-6B/blob/main/web_demo.py
-while mistakes are mine
-"""
+"""Credit to https://github.com/THUDM/ChatGLM2-6B/blob/main/web_demo.py while mistakes are mine."""
 # pylint: disable=broad-exception-caught, redefined-outer-name, missing-function-docstring, missing-module-docstring, too-many-arguments, line-too-long, invalid-name, redefined-builtin, redefined-argument-from-local
 # import gradio as gr
 
@@ -74,7 +72,7 @@ gr.Chatbot.postprocess = postprocess
 
 
 def parse_text(text):
-    """copy from https://github.com/GaiZhenbiao/ChuanhuChatGPT/"""
+    """Copy from https://github.com/GaiZhenbiao/ChuanhuChatGPT/."""
     lines = text.split("\n")
     lines = [line for line in lines if line != ""]
     count = 0
@@ -383,4 +381,4 @@ with gr.Blocks(title="ChatGLM2-6B-int4", theme=gr.themes.Soft(text_size="sm")) a
 # leave one for api access
 # reduce to 3 if OOM
 
-block.queue(concurrency_count=4, max_size=20).launch(debug=True)
+demo.queue(concurrency_count=4, max_size=20).launch(debug=True)