update

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 
 import torch
 from transformers import pipeline, StoppingCriteriaList, MaxTimeCriteria
-import openai
+from openai import OpenAI
 
 from elasticsearch import Elasticsearch
 
@@ -72,7 +72,8 @@ def rag_pipeline(prompt, index="pubmed", num_docs=3, model_name="HuggingFaceH4/z
         openai_model_name = model_name.split('/')[1]
         openai_prompt = '\n\n'.join([m['content'] for m in messages])
 
-        openai_res = openai.Completion.create(model=openai_model_name,
+        client = OpenAI()
+        openai_res = client.completions.create(model=openai_model_name,
                                        prompt=openai_prompt,
                                        max_tokens=generation_kwargs["max_new_tokens"],
                                        n=1,