Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -23,6 +23,7 @@ model = genai.GenerativeModel('models/gemini-1.5-flash-latest')
|
|
23 |
|
24 |
# Load the RAG multi-modal model
|
25 |
RAG = RAGMultiModalModel.from_pretrained("vidore/colpali-v1.2", verbose=1)
|
|
|
26 |
|
27 |
# Specify the index path where the index was saved during the first run
|
28 |
index_path = "/home/mohammadaqib/Desktop/project/research/Multi-Modal-RAG/Colpali/BCC"
|
|
|
23 |
|
24 |
# Load the RAG multi-modal model
|
25 |
RAG = RAGMultiModalModel.from_pretrained("vidore/colpali-v1.2", verbose=1)
|
26 |
+
RAG.to(device)
|
27 |
|
28 |
# Specify the index path where the index was saved during the first run
|
29 |
index_path = "/home/mohammadaqib/Desktop/project/research/Multi-Modal-RAG/Colpali/BCC"
|