Pavan178 commited on
Commit
18a61b7
·
verified ·
1 Parent(s): f74eb2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,9 +7,10 @@ from langchain.vectorstores import FAISS
7
  from langchain.chains import ConversationalRetrievalChain
8
  from langchain.chat_models import ChatOpenAI
9
  from langchain.memory import ConversationBufferMemory
 
10
 
11
  openai_api_key = os.environ.get("OPENAI_API_KEY")
12
-
13
  class AdvancedPdfChatbot:
14
  def __init__(self, openai_api_key):
15
  os.environ["OPENAI_API_KEY"] = openai_api_key
 
7
  from langchain.chains import ConversationalRetrievalChain
8
  from langchain.chat_models import ChatOpenAI
9
  from langchain.memory import ConversationBufferMemory
10
+ import spaces
11
 
12
  openai_api_key = os.environ.get("OPENAI_API_KEY")
13
+ @spaces.GPU
14
  class AdvancedPdfChatbot:
15
  def __init__(self, openai_api_key):
16
  os.environ["OPENAI_API_KEY"] = openai_api_key