Aqib2489 commited on
Commit
0590c46
·
verified ·
1 Parent(s): fd29fdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ genai.configure(api_key=GOOGLE_API_KEY)
22
  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", device_map="cpu", 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"
 
22
  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"