hellokawei commited on
Commit
90be7f3
·
verified ·
1 Parent(s): a138de3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -9
app.py CHANGED
@@ -2,12 +2,14 @@ import gradio as gr
2
  import os
3
  from openai import OpenAI
4
 
 
5
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
6
- openai_client = OpenAI(api_key=OPENAI_API_KEY)
7
- def generate_response(prompt,model,temperature, top_p, max_tokens, repetition_penalty):
 
8
  try:
9
  response = openai_client.chat.completions.create(
10
- model="gpt-3.5-turbo", # or another model of your choice
11
  messages=[{"role": "user", "content": prompt}],
12
  temperature=temperature,
13
  top_p=top_p,
@@ -17,19 +19,19 @@ def generate_response(prompt,model,temperature, top_p, max_tokens, repetition_pe
17
  )
18
  return response.choices[0].message.content.strip()
19
  except Exception as e:
20
- return f"OpenAI API Error: {str(e)}"
21
 
22
  iface = gr.Interface(
23
  fn=generate_response,
24
  inputs=[
25
- gr.Textbox(label="Prompt", lines=6, placeholder="Ask something..."),
26
- gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature"),
27
  gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p"),
28
- gr.Slider(minimum=32, maximum=2048, value=512, step=32, label="Max New Tokens"),
29
- gr.Slider(minimum=1.0, maximum=2.0, value=1.1, step=0.1, label="Repetition Penalty")
30
  ],
31
  outputs="text",
32
- title="🧠 DeepSeek LLM Chat with Parameter Tuning",
33
  theme=gr.themes.Soft()
34
  )
35
 
 
2
  import os
3
  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
+ def generate_response(prompt, model_provider, temperature, top_p, max_tokens, repetition_penalty):
10
  try:
11
  response = openai_client.chat.completions.create(
12
+ model="gpt-3.5-turbo",
13
  messages=[{"role": "user", "content": prompt}],
14
  temperature=temperature,
15
  top_p=top_p,
 
19
  )
20
  return response.choices[0].message.content.strip()
21
  except Exception as e:
22
+ return f"OpenAI API错误: {str(e)}"
23
 
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 聊天演示(参数可调)",
35
  theme=gr.themes.Soft()
36
  )
37