Update app.py

app.py

@@ -1,14 +1,12 @@
-import torch
 import gradio as gr
 import json
 from transformers import pipeline
 
-model_path = pipeline("translation", model="facebook/nllb-200-distilled-600M",torch_dtype=torch.bfloat16)
-# Initialize the pipeline
+# Initialize the pipeline with the model
 try:
-    text_translator = pipeline("translation", model=model_path, torch_dtype=torch.bfloat16)
+    text_translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
 except Exception as e:
-    raise ValueError(f"Error initializing the model. Check the model path: {e}")
+    raise ValueError(f"Error initializing the model: {e}")
 
 # Load the JSON data for language codes
 try:
@@ -19,7 +17,6 @@ except FileNotFoundError:
 except json.JSONDecodeError:
     raise ValueError("Error decoding the JSON file. Ensure it is formatted correctly.")
 
-
 # Create a function to get the FLORES-200 code for a given language
 def get_FLORES_code_from_language(language):
     for entry in language_data:
@@ -27,7 +24,6 @@ def get_FLORES_code_from_language(language):
             return entry['FLORES-200 code']
     return None
 
-
 # Translation function
 def translate_text(text, destination_language):
     if not text.strip():
@@ -38,12 +34,12 @@ def translate_text(text, destination_language):
         return f"Language '{destination_language}' is not supported."
 
     try:
+        # Perform translation
         translation = text_translator(text, src_lang="eng_Latn", tgt_lang=dest_code)
         return translation[0]["translation_text"]
     except Exception as e:
         return f"Error during translation: {e}"
 
-
 # Extract available languages from JSON
 available_languages = [entry['Language'] for entry in language_data]