Spaces:
Runtime error
Runtime error
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
|
| 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)
|