bug fix

app.py

@@ -32,27 +32,30 @@ def translate_text(source_lang, target_lang, input_text, model_dict):
     source_code = flores_codes[source_lang]
     target_code = flores_codes[target_lang]
 
-    model = model_dict[model_name]["model"]
-    tokenizer = model_dict[model_name]["tokenizer"]
-
-    translator = pipeline(
-        "translation",
-        model=model,
-        tokenizer=tokenizer,
-        src_lang=source_code,
-        tgt_lang=target_code,
-    )
-    translated_output = translator(input_text, max_length=400)
-
-    end_time = time.time()
-
-    translated_result = {
-        "inference_time": end_time - start_time,
-        "source": source_lang,
-        "target": target_lang,
-        "result": translated_output[0]["translation_text"],
-    }
-    return translated_result
+    if model_name in model_dict:
+        model = model_dict[model_name]["model"]
+        tokenizer = model_dict[model_name]["tokenizer"]
+
+        translator = pipeline(
+            "translation",
+            model=model,
+            tokenizer=tokenizer,
+            src_lang=source_code,
+            tgt_lang=target_code,
+        )
+        translated_output = translator(input_text, max_length=400)
+
+        end_time = time.time()
+
+        translated_result = {
+            "inference_time": end_time - start_time,
+            "source": source_lang,
+            "target": target_lang,
+            "result": translated_output[0]["translation_text"],
+        }
+        return translated_result
+    else:
+        raise KeyError(f"Model '{model_name}' not found in model_dict")
 
 async def main():
     print("\tInitializing models")
@@ -69,17 +72,17 @@ async def main():
 
     outputs = gr.outputs.JSON()
 
-    title = "The Master Betters Translator"
+    title = "Masterful Translator"
 
     app_description = (
-        "This is a beta version of The Master Betters Translator that utilizes pre-trained language models for translation."
+        "This is a beta version of the Masterful Translator that utilizes pre-trained language models for translation."
     )
     examples = [["English", "Nepali", "Hello, how are you?"]]
 
     gr.Interface(
-        translate_text,
-        inputs,
-        outputs,
+        fn=translate_text,
+        inputs=inputs,
+        outputs=outputs,
         title=title,
         description=app_description,
         examples=examples,