Update app

app.py

@@ -99,7 +99,7 @@ def load_recommender(path, start_page=1):
     return 'Corpus Loaded.'
 
 
-def generate_text(openAI_key, prompt, model_name="text-davinci-003"):
+def generate_text(openAI_key, prompt, question, model="gpt-3.5-turbo"):
     openai.api_key = openAI_key
     if model_name.startswith('gpt'):
         completions = openai.ChatCompletion.create(
@@ -139,10 +139,11 @@ def generate_answer(question, openAI_key, model_name):
               "search results which has nothing to do with the question. Only answer what is asked. The "\
               "answer should be short and concise. "
     
-    prompt += f"Query: {question}\nAnswer:"
+    
     if model_name.startswith('gpt'):
         answer = generate_text(openAI_key, prompt, question, model_name)
     else:
+        prompt += f"Query: {question}\nAnswer:"
         answer = generate_text(openAI_key, prompt, model_name)
     return answer