Ubuntu
commited on
Commit
·
44943ab
1
Parent(s):
82d9320
remove hugggingface auth login block
Browse files- gateway/main.py +5 -8
- 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 |
-
|
| 108 |
-
|
| 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 |
-
|
| 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 |
-
|
| 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 |
-
|
| 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,
|
|
|
|
|
|
|
|
|
|
| 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.")
|