EdmundYi commited on
Commit
51f5a1e
·
verified ·
1 Parent(s): 1e04c7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,10 +5,10 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
5
  openai_client = OpenAI(api_key=OPENAI_API_KEY)
6
  DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
7
  deepseek_client = OpenAI(api_key=DEEPSEEK_API_KEY, base_url="https://api.deepseek.com")
8
- def generate_response(prompt, temperature, top_p, max_tokens, repetition_penalty):
9
  try:
10
  response = openai_client.chat.completions.create(
11
- model="gpt-3.5-turbo", # model="deepseek-chat"
12
  messages=[{"role": "user", "content": prompt}],
13
  temperature=temperature,
14
  top_p=top_p,
 
5
  openai_client = OpenAI(api_key=OPENAI_API_KEY)
6
  DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
7
  deepseek_client = OpenAI(api_key=DEEPSEEK_API_KEY, base_url="https://api.deepseek.com")
8
+ def generate_response(prompt, model_provider, temperature, top_p, max_tokens, repetition_penalty):
9
  try:
10
  response = openai_client.chat.completions.create(
11
+ model="deepseek-chat", # model="gpt-3.5-turbo"
12
  messages=[{"role": "user", "content": prompt}],
13
  temperature=temperature,
14
  top_p=top_p,