comment uri

README.md

@@ -9,3 +9,4 @@ license: bsd
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+    

docker-compose.yaml

@@ -0,0 +1,18 @@
+version: "3.8"
+
+services:
+  app:
+    build:
+      context: .
+    ports:
+      - "7860:8000"
+    environment:
+      - HF_HOME=/app/transformers_cache
+      - TRANSFORMERS_CACHE=/app/transformers_cache
+      - HF_DATASETS_CACHE=/app/datasets_cache
+      - TOKENIZERS_PARALLELISM=true
+    volumes:
+      - ./transformers_cache:/app/transformers_cache
+      - ./datasets_cache:/app/datasets_cache
+      - ./app:/app/
+      - ./deepspeech-ht/model:/app/deepspeech/model

main.py

@@ -11,7 +11,6 @@ app = FastAPI(docs_url="/")
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 
-
 class TranslationRequest(BaseModel):
     user_input: str
     source_lang: str
@@ -27,6 +26,10 @@ def load_model(pretrained_model: str = "facebook/m2m100_1.2B", cache_dir: str =
 
 @app.post("/translate")
 async def translate(request: TranslationRequest):
+    """
+    language support
+    Afrikaans (af), Amharic (am), Arabic (ar), Asturian (ast), Azerbaijani (az), Bashkir (ba), Belarusian (be), Bulgarian (bg), Bengali (bn), Breton (br), Bosnian (bs), Catalan; Valencian (ca), Cebuano (ceb), Czech (cs), Welsh (cy), Danish (da), German (de), Greeek (el), English (en), Spanish (es), Estonian (et), Persian (fa), Fulah (ff), Finnish (fi), French (fr), Western Frisian (fy), Irish (ga), Gaelic; Scottish Gaelic (gd), Galician (gl), Gujarati (gu), Hausa (ha), Hebrew (he), Hindi (hi), Croatian (hr), Haitian; Haitian Creole (ht), Hungarian (hu), Armenian (hy), Indonesian (id), Igbo (ig), Iloko (ilo), Icelandic (is), Italian (it), Japanese (ja), Javanese (jv), Georgian (ka), Kazakh (kk), Central Khmer (km), Kannada (kn), Korean (ko), Luxembourgish; Letzeburgesch (lb), Ganda (lg), Lingala (ln), Lao (lo), Lithuanian (lt), Latvian (lv), Malagasy (mg), Macedonian (mk), Malayalam (ml), Mongolian (mn), Marathi (mr), Malay (ms), Burmese (my), Nepali (ne), Dutch; Flemish (nl), Norwegian (no), Northern Sotho (ns), Occitan (post 1500) (oc), Oriya (or), Panjabi; Punjabi (pa), Polish (pl), Pushto; Pashto (ps), Portuguese (pt), Romanian; Moldavian; Moldovan (ro), Russian (ru), Sindhi (sd), Sinhala; Sinhalese (si), Slovak (sk), Slovenian (sl), Somali (so), Albanian (sq), Serbian (sr), Swati (ss), Sundanese (su), Swedish (sv), Swahili (sw), Tamil (ta), Thai (th), Tagalog (tl), Tswana (tn), Turkish (tr), Ukrainian (uk), Urdu (ur), Uzbek (uz), Vietnamese (vi), Wolof (wo), Xhosa (xh), Yiddish (yi), Yoruba (yo), Chinese (zh), Zulu (zu)
+    """
     time_start = time.time()
     tokenizer, model = load_model()
     src_lang = request.source_lang