Update main.py

main.py

@@ -12,7 +12,7 @@ import os
 app = Flask(__name__)
 app.json.sort_keys = False
 
-parser = argparse.ArgumentParser(description="An example of Qwen demo with a similar API to OAI.")
+parser = argparse.ArgumentParser(description="An example of Zhipu GLM-4 with a similar API to OAI.")
 parser.add_argument("--host", type=str, help="Set the ip address.(default: 0.0.0.0)", default='0.0.0.0')
 parser.add_argument("--port", type=int, help="Set the port.(default: 7860)", default=7860)
 args = parser.parse_args()
@@ -27,7 +27,7 @@ def model_list():
         "object": "list",
         "data": [
             {
-                "id": "qwen",
+                "id": "chatglm4",
                 "object": "model",
                 "created": time_now,
                 "owned_by": "tastypear"
@@ -44,7 +44,7 @@ def model_list():
 
 @app.route("/", methods=["GET"])
 def index():
-    return Response(f'QW1_5 OpenAI Compatible API<br><br>'+
+    return Response(f'GLM4 OpenAI Compatible API<br><br>'+
         f'Set "{os.getenv("SPACE_URL")}/api" as proxy (or API Domain) in your Chatbot.<br><br>'+
         f'The complete API is: {os.getenv("SPACE_URL")}/api/v1/chat/completions')
 
@@ -136,7 +136,7 @@ def gen_res_data(data, time_now=0, start=False):
         "id": "chatcmpl",
         "object": "chat.completion.chunk",
         "created": time_now,
-        "model": "qwen1_5",
+        "model": "chatglm4",
         "choices": [{"index": 0, "finish_reason": None}],
     }