Update app.py

app.py

@@ -168,7 +168,7 @@ def run_query(input_query, country, model_sel):
     if model_sel == "chatGPT":
         # res = pipe.run(query=input_text, documents=docs)
         res = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": get_prompt(docs, input_query)}])
-        output = res["results"][0]
+        output = res.choices[0].message.content
         references = get_refs(docs, output)
     # else:
     #     res = client.text_generation(get_prompt_llama2(docs, query=input_query), max_new_tokens=4000, temperature=0.01, model=model)