hellokawei commited on
Commit
61a9bbf
·
verified ·
1 Parent(s): fd49dcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -6,6 +6,9 @@ from openai import OpenAI
6
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
7
  openai_client = OpenAI(api_key=OPENAI_API_KEY) # 修正这里!
8
 
 
 
 
9
  def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty):
10
  try:
11
  response = openai_client.chat.completions.create(
@@ -24,11 +27,12 @@ def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty
24
  iface = gr.Interface(
25
  fn=generate_response,
26
  inputs=[
27
- gr.Textbox(label="输入提示", lines=6, placeholder="在这里输入问题..."),
28
- gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="温度 (Temperature)"),
 
29
  gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p"),
30
- gr.Slider(minimum=32, maximum=2048, value=512, step=32, label="最大生成长度 (Max Tokens)"),
31
- gr.Slider(minimum=1.0, maximum=2.0, value=1.1, step=0.1, label="重复惩罚 (Repetition Penalty)")
32
  ],
33
  outputs="text",
34
  title="🧠 DeepSeek LLM 聊天演示(参数可调)",
 
6
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
7
  openai_client = OpenAI(api_key=OPENAI_API_KEY) # 修正这里!
8
 
9
+ DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
10
+ deepseek_client = OpenAI(api_key=DEEPSEEK_API_KEY, base_url="https://api.deepseek.com")
11
+
12
  def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty):
13
  try:
14
  response = openai_client.chat.completions.create(
 
27
  iface = gr.Interface(
28
  fn=generate_response,
29
  inputs=[
30
+ gr.Dropdown(choices=["DeepSeek", "OpenAI"], value="DeepSeek", label="Model Provider"),
31
+ gr.Textbox(label="Prompt", lines=6, placeholder="Ask something..."),
32
+ gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature"),
33
  gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p"),
34
+ gr.Slider(minimum=32, maximum=2048, value=512, step=32, label="Max New Tokens"),
35
+ gr.Slider(minimum=1.0, maximum=2.0, value=1.1, step=0.1, label="Repetition Penalty")
36
  ],
37
  outputs="text",
38
  title="🧠 DeepSeek LLM 聊天演示(参数可调)",