tastypear commited on
Commit
c511634
1 Parent(s): 6d2b530

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -12,7 +12,7 @@ import os
12
  app = Flask(__name__)
13
  app.json.sort_keys = False
14
 
15
- parser = argparse.ArgumentParser(description="An example of Qwen demo with a similar API to OAI.")
16
  parser.add_argument("--host", type=str, help="Set the ip address.(default: 0.0.0.0)", default='0.0.0.0')
17
  parser.add_argument("--port", type=int, help="Set the port.(default: 7860)", default=7860)
18
  args = parser.parse_args()
@@ -28,7 +28,7 @@ def model_list():
28
  "object": "list",
29
  "data": [
30
  {
31
- "id": "qwen",
32
  "object": "model",
33
  "created": time_now,
34
  "owned_by": "tastypear"
@@ -158,7 +158,7 @@ def gen_res_data(data, time_now=0, start=False):
158
  "id": "chatcmpl",
159
  "object": "chat.completion.chunk",
160
  "created": time_now,
161
- "model": "qwen2_5",
162
  "choices": [{"index": 0, "finish_reason": None}],
163
  }
164
 
 
12
  app = Flask(__name__)
13
  app.json.sort_keys = False
14
 
15
+ parser = argparse.ArgumentParser(description="An example of Hunyuan demo with a similar API to OAI.")
16
  parser.add_argument("--host", type=str, help="Set the ip address.(default: 0.0.0.0)", default='0.0.0.0')
17
  parser.add_argument("--port", type=int, help="Set the port.(default: 7860)", default=7860)
18
  args = parser.parse_args()
 
28
  "object": "list",
29
  "data": [
30
  {
31
+ "id": "hunyuan-large",
32
  "object": "model",
33
  "created": time_now,
34
  "owned_by": "tastypear"
 
158
  "id": "chatcmpl",
159
  "object": "chat.completion.chunk",
160
  "created": time_now,
161
+ "model": "hunyuan-large",
162
  "choices": [{"index": 0, "finish_reason": None}],
163
  }
164