Update app.py

app.py

@@ -4,11 +4,14 @@ import torch
 from transformers import pipeline
 
 # Initialize the translation pipeline
-translator = pipeline(
-    "translation",
-    model="Helsinki-NLP/opus-mt-tc-big-en-es",
-    device=0 if torch.cuda.is_available() else -1,  # Use GPU if available, otherwise CPU
-)
+def initialize_translator():
+    try:
+        device = 0 if torch.cuda.is_available() else -1  # Use GPU if available, otherwise CPU
+        return pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-en-es", device=device)
+    except Exception as e:
+        raise RuntimeError(f"Failed to initialize translation pipeline: {e}")
+
+translator = initialize_translator()
 
 # Define the custom glossary
 custom_glossary = {
@@ -151,3 +154,4 @@ iface = gr.Interface(
 # Launch the interface
 if __name__ == "__main__":
     iface.launch()
+