added russian

app.py

@@ -91,8 +91,11 @@ def fix_string(batch):
 # tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-uig-script_arabic")
 # model = VitsModel.from_pretrained("piyazon/AliKurban")
 # tokenizer = AutoTokenizer.from_pretrained("piyazon/AliKurban")
-model = VitsModel.from_pretrained("piyazon/qutadgu_bilik")
-tokenizer = AutoTokenizer.from_pretrained("piyazon/qutadgu_bilik")
+model_ug = VitsModel.from_pretrained("piyazon/qutadgu_bilik")
+tokenizer_ug = AutoTokenizer.from_pretrained("piyazon/qutadgu_bilik")
+
+model_ru = VitsModel.from_pretrained("facebook/mms-tts-rus")
+tokenizer_ru = AutoTokenizer.from_pretrained("facebook/mms-tts-rus")
 
 # Pydantic model for request body
 class TextInput(BaseModel):
@@ -101,14 +104,21 @@ class TextInput(BaseModel):
 """
 curl -X POST https://piyazon-tts-piyazon.hf.space/generate-tts \
     -H "Content-Type: application/json" \
-    -d '{"text": "Hello, world!"}' \
+    -d '{"text": "Hello, world!", "lang":"ug"}' \
     --output output.wav
 """
 @app.post("/generate-tts")
 async def generate_tts(input: TextInput):
     try:
+        if input.lang=="ug":
+            model = model_ug
+            tokenizer = tokenizer_ug
+            inputs = tokenizer(fix_string(input.text), return_tensors="pt")
+        else:
+            model = model_ru
+            tokenizer = tokenizer_ru
+            inputs = tokenizer(input.text, return_tensors="pt")
         # Tokenize input text
-        inputs = tokenizer(fix_string(input.text), return_tensors="pt")
         
         # Generate waveform
         with torch.no_grad():