Tbb1111 commited on
Commit
284837e
·
verified ·
1 Parent(s): e8dce74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
3
  import gradio as gr
4
 
5
- model_name = "deepseek-ai/deepseek-llm-7b-chat"
6
 
7
  # 加载 tokenizer 和模型
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -42,7 +42,7 @@ iface = gr.Interface(
42
  gr.Slider(0.1, 1.0, step=0.05, value=0.9, label="Top-p")
43
  ],
44
  outputs=gr.Textbox(label="模型回复"),
45
- title="DeepSeek LLM 7B Chat 演示",
46
  description="基于 Hugging Face Spaces 的部署示例"
47
  )
48
 
 
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
3
  import gradio as gr
4
 
5
+ model_name = "deepseek-ai/deepseek-vl-1.3b-chat"
6
 
7
  # 加载 tokenizer 和模型
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
42
  gr.Slider(0.1, 1.0, step=0.05, value=0.9, label="Top-p")
43
  ],
44
  outputs=gr.Textbox(label="模型回复"),
45
+ title="DeepSeekChat 演示",
46
  description="基于 Hugging Face Spaces 的部署示例"
47
  )
48