Update app.py

app.py

@@ -34,6 +34,7 @@ for lang, model_name in translation_models.items():
     try:
         loaded_translation_models[lang] = MarianMTModel.from_pretrained(model_name)
         loaded_translation_tokenizers[lang] = MarianTokenizer.from_pretrained(model_name)
+        print(f"Successfully loaded translation model for {lang}")
     except Exception as e:
         print(f"Error loading model for {lang}: {e}")
 
@@ -64,7 +65,9 @@ def translate_text(text, target_lang="Spanish"):
 # Combined Captioning and Translation function
 def caption_and_translate(image, target_lang="Spanish"):
     caption_text = caption(image)
+    print(f"Generated caption: {caption_text}")
     translated_caption = translate_text(caption_text, target_lang)
+    print(f"Translated caption: {translated_caption}")
     return caption_text, translated_caption
 
 # Create Gradio interfaces
@@ -91,3 +94,4 @@ final_interface = gr.TabbedInterface([interface1, interface2, interface3],
                                      title=title, theme=gr.themes.Soft())
 
 final_interface.launch(inbrowser=True)
+