Update app.py

app.py

@@ -5,12 +5,13 @@ import torch
 
 @st.cache_resource
 def load_all_models():
-    # Load Tamil ➝ English translation model
+    # Load Tamil ➝ English translation model (with remote code)
     translation_pipeline = pipeline(
         "translation",
         model="ai4bharat/indictrans2-indic-en-dist-200M",
         tokenizer="ai4bharat/indictrans2-indic-en-dist-200M",
         src_lang="ta", tgt_lang="en",
+        trust_remote_code=True,  # 🔥 Must add this
         device=0 if torch.cuda.is_available() else -1,
     )
 
@@ -24,6 +25,7 @@ def load_all_models():
 
     return translation_pipeline, img_pipe
 
+
 def main():
     st.set_page_config(page_title="Tamil ➝ English ➝ Image", layout="centered")
     st.title("🌐 Tamil to English to Image Generator")