rivapereira123 commited on
Commit
03da98c
·
verified ·
1 Parent(s): a13a659

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -286,10 +286,9 @@ class GazaKnowledgeBase:
286
  logger.info("Initializing Gaza Knowledge Base...")
287
 
288
  # Initialize embedding model
289
- self.embedding_model = SentenceTransformer(
290
- 'sentence-transformers/all-MiniLM-L6-v2',
291
- device='cpu' # Use CPU for better compatibility
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()