Ubuntu commited on
Commit
44943ab
·
1 Parent(s): 82d9320

remove hugggingface auth login block

Browse files
Files changed (2) hide show
  1. gateway/main.py +5 -8
  2. utils/model.py +6 -5
gateway/main.py CHANGED
@@ -104,8 +104,8 @@ async def tts(
104
  ):
105
  lang = lang.lower() if lang else "hindi"
106
 
107
- if lang == "marathi":
108
- text = " " + text + " "
109
 
110
  for language, lang_code in Language_CODE.items():
111
  if language == lang:
@@ -123,8 +123,7 @@ async def tts(
123
  processed = transliterate(normalize(text))
124
  reference_text = transliterate(asr(ref_path, language_id))
125
 
126
- if lang == "marathi":
127
- reference_text = reference_text + " "
128
 
129
  output_file = f"{uuid.uuid4()}.wav"
130
 
@@ -145,8 +144,7 @@ async def get_inference(
145
 
146
  lang = lang.lower() if lang else "hindi"
147
 
148
- if lang == "marathi":
149
- text = " " + text + " "
150
 
151
  for language, lang_code in Language_CODE.items():
152
  if language == lang:
@@ -166,8 +164,7 @@ async def get_inference(
166
  processed = transliterate(normalize(text))
167
  reference_text = transliterate(asr(ref_path, language_id))
168
 
169
- if lang == "marathi":
170
- reference_text = reference_text + " "
171
 
172
  output_file = f"{uuid.uuid4()}.wav"
173
 
 
104
  ):
105
  lang = lang.lower() if lang else "hindi"
106
 
107
+
108
+ text = " " + text + " "
109
 
110
  for language, lang_code in Language_CODE.items():
111
  if language == lang:
 
123
  processed = transliterate(normalize(text))
124
  reference_text = transliterate(asr(ref_path, language_id))
125
 
126
+ reference_text = reference_text + " "
 
127
 
128
  output_file = f"{uuid.uuid4()}.wav"
129
 
 
144
 
145
  lang = lang.lower() if lang else "hindi"
146
 
147
+ text = " " + text + " "
 
148
 
149
  for language, lang_code in Language_CODE.items():
150
  if language == lang:
 
164
  processed = transliterate(normalize(text))
165
  reference_text = transliterate(asr(ref_path, language_id))
166
 
167
+ reference_text = reference_text + " "
 
168
 
169
  output_file = f"{uuid.uuid4()}.wav"
170
 
utils/model.py CHANGED
@@ -1,6 +1,5 @@
1
  import torch
2
  from transformers import AutoModel
3
- from huggingface_hub import login
4
 
5
 
6
  class ModelManager:
@@ -12,12 +11,13 @@ class ModelManager:
12
 
13
  def __enter__(self):
14
  print("Loading model into memory...")
15
-
16
- if self.hf_token:
17
- login(token=self.hf_token)
18
 
19
  self.model = AutoModel.from_pretrained(
20
- self.repo_id, ignore_mismatched_sizes=True, trust_remote_code=True
 
 
 
21
  ).to(self.device)
22
 
23
  print("Model loaded and ready!")
@@ -28,4 +28,5 @@ class ModelManager:
28
 
29
  del self.model
30
  torch.cuda.empty_cache()
 
31
  print("Model unloaded successfully.")
 
1
  import torch
2
  from transformers import AutoModel
 
3
 
4
 
5
  class ModelManager:
 
11
 
12
  def __enter__(self):
13
  print("Loading model into memory...")
14
+ print("HF Token received:", "YES" if self.hf_token else "NO")
 
 
15
 
16
  self.model = AutoModel.from_pretrained(
17
+ self.repo_id,
18
+ ignore_mismatched_sizes=True,
19
+ trust_remote_code=True,
20
+ token=self.hf_token
21
  ).to(self.device)
22
 
23
  print("Model loaded and ready!")
 
28
 
29
  del self.model
30
  torch.cuda.empty_cache()
31
+
32
  print("Model unloaded successfully.")