Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,8 +37,7 @@ def pdf_changes(pdf_doc):
|
|
| 37 |
return "Ready"
|
| 38 |
|
| 39 |
def book_changes(book):
|
| 40 |
-
|
| 41 |
-
db = torch.load(book, map_location='cpu')
|
| 42 |
llm=HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":1, "max_length":1000000})
|
| 43 |
global qa
|
| 44 |
qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever(search_kwargs={"k": 3}))
|
|
|
|
| 37 |
return "Ready"
|
| 38 |
|
| 39 |
def book_changes(book):
|
| 40 |
+
db = torch.load(book, map_location=torch.device('cpu'))
|
|
|
|
| 41 |
llm=HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":1, "max_length":1000000})
|
| 42 |
global qa
|
| 43 |
qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever(search_kwargs={"k": 3}))
|