Spaces:
Sleeping
Sleeping
Commit ·
70de5e9
1
Parent(s): f70514c
Fixes
Browse files
src/utils/data_processor.py
CHANGED
|
@@ -88,7 +88,7 @@ class TranslationService:
|
|
| 88 |
self.api_token = self._get_hf_token()
|
| 89 |
# Using Helsinki-NLP model - better Inference API support for Hindi→English
|
| 90 |
self.translation_model = "Helsinki-NLP/opus-mt-hi-en"
|
| 91 |
-
self.base_url = "https://
|
| 92 |
logger.info("Initialized HF Inference API for translation")
|
| 93 |
|
| 94 |
def _get_hf_token(self) -> Optional[str]:
|
|
@@ -114,7 +114,8 @@ class TranslationService:
|
|
| 114 |
|
| 115 |
try:
|
| 116 |
headers = {"Authorization": f"Bearer {self.api_token}"}
|
| 117 |
-
|
|
|
|
| 118 |
|
| 119 |
# Simplified payload for Helsinki-NLP models
|
| 120 |
payload = {"inputs": text}
|
|
|
|
| 88 |
self.api_token = self._get_hf_token()
|
| 89 |
# Using Helsinki-NLP model - better Inference API support for Hindi→English
|
| 90 |
self.translation_model = "Helsinki-NLP/opus-mt-hi-en"
|
| 91 |
+
self.base_url = "https://router.huggingface.co"
|
| 92 |
logger.info("Initialized HF Inference API for translation")
|
| 93 |
|
| 94 |
def _get_hf_token(self) -> Optional[str]:
|
|
|
|
| 114 |
|
| 115 |
try:
|
| 116 |
headers = {"Authorization": f"Bearer {self.api_token}"}
|
| 117 |
+
# New router API format: /models/{model_id}
|
| 118 |
+
url = f"{self.base_url}/models/{self.translation_model}"
|
| 119 |
|
| 120 |
# Simplified payload for Helsinki-NLP models
|
| 121 |
payload = {"inputs": text}
|