Update app.py

app.py

@@ -13,9 +13,10 @@ model_list = [
 lang_list = list(lang_ids.keys())
 
 def translate_to_english(text, base_lang):
-    if base_lang == "en":
+    if base_lang == "English":
         return text
     else:
+        base_lang = lang_ids[base_lang]
         new_text = translate("facebook/m2m100_418M", text, base_lang, "en")
         return new_text
     
@@ -29,6 +30,7 @@ def biogpt(
 ):
     
     en_prompt = translate_to_english(prompt, base_lang)
+    
     model = BioGptForCausalLM.from_pretrained(model_id)
     tokenizer = BioGptTokenizer.from_pretrained(model_id)
     generator = pipeline('text-generation', model=model, tokenizer=tokenizer)