miaoyibo commited on
Commit
3f5215b
·
1 Parent(s): 790da55
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -30,8 +30,8 @@ from kimi_vl.serve.chat_utils import (
30
  from kimi_vl.serve.inference import kimi_vl_generate, load_model
31
  from kimi_vl.serve.examples import get_examples
32
 
33
- TITLE = """<h1 align="left" style="min-width:200px; margin-top:0;">Chat with Kimi-VL-A3B-Thinking🤔 </h1>"""
34
- DESCRIPTION_TOP = """<a href="https://github.com/MoonshotAI/Kimi-VL" target="_blank">Kimi-VL-A3B-Thinking</a> is a multi-modal LLM that can understand text and images, and generate text with thinking processes. For non-thinking version, please try [Kimi-VL-A3B](https://huggingface.co/spaces/moonshotai/Kimi-VL-A3B)."""
35
  DESCRIPTION = """"""
36
  ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
37
  DEPLOY_MODELS = dict()
@@ -40,7 +40,7 @@ logger = configure_logger()
40
 
41
  def parse_args():
42
  parser = argparse.ArgumentParser()
43
- parser.add_argument("--model", type=str, default="Kimi-VL-A3B-Thinking")
44
  parser.add_argument(
45
  "--local-path",
46
  type=str,
@@ -252,7 +252,7 @@ def build_demo(args: argparse.Namespace) -> gr.Blocks:
252
  with gr.Column(scale=4):
253
  with gr.Row():
254
  chatbot = gr.Chatbot(
255
- elem_id="Kimi-VL-A3B-Thinking-chatbot",
256
  show_share_button=True,
257
  bubble_full_width=False,
258
  height=600,
@@ -330,7 +330,7 @@ def build_demo(args: argparse.Namespace) -> gr.Blocks:
330
  del_last_btn.click(delete_last_conversation, [chatbot, history], output_widgets, show_progress=True)
331
  cancel_btn.click(cancel_outputing, [], [status_display], cancels=predict_events)
332
 
333
- demo.title = "Kimi-VL-A3B-Thinking Chatbot"
334
  return demo
335
 
336
 
@@ -349,4 +349,5 @@ def main(args: argparse.Namespace):
349
 
350
  if __name__ == "__main__":
351
  args = parse_args()
 
352
  main(args)
 
30
  from kimi_vl.serve.inference import kimi_vl_generate, load_model
31
  from kimi_vl.serve.examples import get_examples
32
 
33
+ TITLE = """<h1 align="left" style="min-width:200px; margin-top:0;">Chat with Kimi-Dev-72B🤔 </h1>"""
34
+ DESCRIPTION_TOP = """<a href="https://github.com/MoonshotAI/Kimi-VL" target="_blank">Kimi-Dev-72B</a> is a multi-modal LLM that can understand text and images, and generate text with thinking processes. For non-thinking version, please try [Kimi-VL-A3B](https://huggingface.co/spaces/moonshotai/Kimi-VL-A3B)."""
35
  DESCRIPTION = """"""
36
  ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
37
  DEPLOY_MODELS = dict()
 
40
 
41
  def parse_args():
42
  parser = argparse.ArgumentParser()
43
+ parser.add_argument("--model", type=str, default="Kimi-Dev-72B")
44
  parser.add_argument(
45
  "--local-path",
46
  type=str,
 
252
  with gr.Column(scale=4):
253
  with gr.Row():
254
  chatbot = gr.Chatbot(
255
+ elem_id="Kimi-Dev-72B",
256
  show_share_button=True,
257
  bubble_full_width=False,
258
  height=600,
 
330
  del_last_btn.click(delete_last_conversation, [chatbot, history], output_widgets, show_progress=True)
331
  cancel_btn.click(cancel_outputing, [], [status_display], cancels=predict_events)
332
 
333
+ demo.title = "Kimi-Dev-72B"
334
  return demo
335
 
336
 
 
349
 
350
  if __name__ == "__main__":
351
  args = parse_args()
352
+ print(args)
353
  main(args)