Update apis/chat_api.py

apis/chat_api.py

@@ -148,11 +148,11 @@ class ChatAPIApp:
         )
         from_language: str = Field(
             default="en",
-            description="(str) `Detect`",
+            description="(str) `translate from`",
         )
         to_language: str = Field(
             default="fa",
-            description="(str) `en`",
+            description="(str) `translate to`",
         )
         input_text: str = Field(
             default="Hello",
@@ -178,7 +178,10 @@ class ChatAPIApp:
         read_model = AutoModelForSeq2SeqLM.from_pretrained(real_name)
         tokenizer = AutoTokenizer.from_pretrained(real_name)
         #translator = pipeline("translation", model=read_model, tokenizer=tokenizer, src_lang=item.from_language, tgt_lang=item.to_language)
-        translator = pipeline("translation", model=real_name, src_lang=item.from_language, tgt_lang=item.to_language)
+        translate_query = (
+            f"translation_{item.from_language}_to_{item.to_language}"
+        )
+        translator = pipeline(translate_query)
         result = translator(item.input_text)    
            
         item_response = {