Update app.py
Browse files
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="
|
| 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,
|