Spaces:
Sleeping
Sleeping
Commit
•
30f357e
1
Parent(s):
d05ba06
cpu
Browse files- app.py +2 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -1,3 +1,5 @@
|
|
|
|
|
|
|
|
| 1 |
import chainlit as cl
|
| 2 |
from faissdenseretrieval import initialize_documents, initialize_faiss_document_store, initialize_rag_pipeline
|
| 3 |
import os
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
torch.device('cpu')
|
| 3 |
import chainlit as cl
|
| 4 |
from faissdenseretrieval import initialize_documents, initialize_faiss_document_store, initialize_rag_pipeline
|
| 5 |
import os
|
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
farm-haystack[faiss]
|
| 2 |
chainlit==0.7.0
|
| 3 |
openai==0.28.0
|
| 4 |
python-dotenv==1.0.0
|
|
|
|
| 1 |
+
farm-haystack[faiss,inference]
|
| 2 |
chainlit==0.7.0
|
| 3 |
openai==0.28.0
|
| 4 |
python-dotenv==1.0.0
|