Update app.py

app.py

@@ -15,7 +15,7 @@ openai.api_model2 = os.environ.get("OPENAI_API_MODEL2")
 BASE_SYSTEM_MESSAGE = """"""
 
 
-def make_prediction(prompt, model, max_tokens=None, temperature=None, top_p=None, top_k=None, repetition_penalty=None, api_key, api_base):
+def make_prediction(prompt, api_model, api_key, api_base, max_tokens=None, temperature=None, top_p=None, top_k=None, repetition_penalty=None):
     completion = openai.Completion.create(
         model=model,
         api_key=api_key,
@@ -52,7 +52,7 @@ def user_double(message, history1, history2):
     history2.append([message, ""])
     return "", history1, history2
 
-def chat(model, history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
+def chat(api_model, history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty, api_key, api_base):
     history = history or []
 
     messages = BASE_SYSTEM_MESSAGE + system_message.strip() + "\n" + \
@@ -65,7 +65,9 @@ def chat(model, history, system_message, max_tokens, temperature, top_p, top_k,
 
     prediction = make_prediction(
         messages,
-        model,
+        api_model,
+        api_key,
+        api_base,
         max_tokens=max_tokens,
         temperature=temperature,
         top_p=top_p,