parthnuwal7 commited on
Commit
70de5e9
·
1 Parent(s): f70514c
Files changed (1) hide show
  1. src/utils/data_processor.py +3 -2
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://api-inference.huggingface.co/models"
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
- url = f"{self.base_url}/{self.translation_model}"
 
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}