Davy592 commited on
Commit
da1e99f
·
1 Parent(s): 4eee0db

Removed device from .encode

Browse files
nygaardcodecommentclassification/api/models.py CHANGED
@@ -286,7 +286,7 @@ class ModelPredictor:
286
  raise ValueError(f"Embedder not loaded for {language}")
287
 
288
  # Encode texts to dense embeddings (no progress bar for API use)
289
- embeddings = embedder.encode(texts, show_progress_bar=False,device="cpu")
290
 
291
  # Run CatBoost prediction on embeddings
292
  return model.predict(embeddings)
 
286
  raise ValueError(f"Embedder not loaded for {language}")
287
 
288
  # Encode texts to dense embeddings (no progress bar for API use)
289
+ embeddings = embedder.encode(texts, show_progress_bar=False)
290
 
291
  # Run CatBoost prediction on embeddings
292
  return model.predict(embeddings)