Update app

app.py

@@ -99,11 +99,11 @@ def load_recommender(path, start_page=1):
     return 'Corpus Loaded.'
 
 
-def generate_text(openAI_key,prompt, engine="text-davinci-003"):
+def generate_text(openAI_key, prompt, model_name="text-davinci-003"):
     openai.api_key = openAI_key
-    if model.startswith('gpt'):
+    if model_name.startswith('gpt'):
         completions = openai.ChatCompletion.create(
-            model=model,
+            model=model_name,
             messages=[
                 {"role": "system", "content": prompt},
                 {"role": "user", "content": question},
@@ -112,7 +112,7 @@ def generate_text(openAI_key,prompt, engine="text-davinci-003"):
         message = completions['choices'][0]['message']['content']
     else:
         completions = openai.Completion.create(
-            engine=engine,
+            engine=model_name,
             prompt=prompt,
             max_tokens=512,
             n=1,
@@ -140,7 +140,7 @@ def generate_answer(question, openAI_key, model_name):
               "answer should be short and concise. \n\nQuery: {question}\nAnswer: "
     
     prompt += f"Query: {question}\nAnswer:"
-    answer = generate_text(openAI_key, prompt,"text-davinci-003")
+    answer = generate_text(openAI_key, prompt, model_name)
     return answer
 
 
@@ -200,4 +200,4 @@ with gr.Blocks() as demo:
 
         btn.click(question_answer, inputs=[url, file, question, openAI_key, model_name], outputs=[answer])
 #openai.api_key = os.getenv('Your_Key_Here') 
-demo.launch()
+demo.launch(share=True)