Update app.py

app.py

@@ -9,6 +9,7 @@ from langchain.docstore.document import Document
 from langchain.document_loaders import PyPDFLoader
 import gradio as gr
 import openai
+import backoff
 
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
@@ -79,6 +80,8 @@ qa = ConversationalRetrievalChain.from_llm(ChatOpenAI(
     temperature=0), docsearch.as_retriever(), return_source_documents=True)
 
 
+@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.error.APIConnectionError)
 def predict(query):
     response = openai.ChatCompletion.create(
         model="gpt-4",