Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -286,10 +286,9 @@ class GazaKnowledgeBase:
|
|
286 |
logger.info("Initializing Gaza Knowledge Base...")
|
287 |
|
288 |
# Initialize embedding model
|
289 |
-
|
290 |
-
|
291 |
-
|
292 |
-
)
|
293 |
|
294 |
# Load or create vector store
|
295 |
self._load_or_create_vector_store()
|
|
|
286 |
logger.info("Initializing Gaza Knowledge Base...")
|
287 |
|
288 |
# Initialize embedding model
|
289 |
+
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
290 |
+
self.embedding_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2',device=device)
|
291 |
+
|
|
|
292 |
|
293 |
# Load or create vector store
|
294 |
self._load_or_create_vector_store()
|