hellokawei commited on
Commit
d492933
·
verified ·
1 Parent(s): a37112e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -4,12 +4,12 @@ from openai import OpenAI
4
 
5
  # 从环境变量读取API密钥
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(
15
  model="gpt-3.5-turbo",
@@ -27,12 +27,11 @@ def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty
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 聊天演示(参数可调)",
 
4
 
5
  # 从环境变量读取API密钥
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): # 现在明确有5个参数
13
  try:
14
  response = openai_client.chat.completions.create(
15
  model="gpt-3.5-turbo",
 
27
  iface = gr.Interface(
28
  fn=generate_response,
29
  inputs=[
30
+ gr.Textbox(label="输入提示", lines=6, placeholder="请输入问题..."),
31
+ gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="温度参数"),
32
+ gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p采样"),
33
+ gr.Slider(minimum=32, maximum=2048, value=512, step=32, label="最大生成长度"),
34
+ gr.Slider(minimum=1.0, maximum=2.0, value=1.1, step=0.1, label="重复惩罚系数")
 
35
  ],
36
  outputs="text",
37
  title="🧠 DeepSeek LLM 聊天演示(参数可调)",