Update app.py

app.py

@@ -78,10 +78,10 @@ def summarize_description(full_description, language):
         return summarization_pipeline(full_description, max_length=150, min_length=50, do_sample=False)[0]['summary_text']  # Summarize in English
 
 # Function to translate the caption and classification result
-def translate_results(caption, top_label, top_prob, language):
+def translate_results(caption, top_label, top_prob, landmarks_dict, language):
     if language == 'ar':
         caption_translated = translation_pipeline(caption, src_lang='eng_Latn', tgt_lang='arb_Arab')[0]['translation_text']  # Translate caption to Arabic
-        classification_result = translation_pipeline(f"أفضل مطابقة: {top_label} باحتمالية {top_prob:.4f}", src_lang='eng_Latn', tgt_lang='arb_Arab')[0]['translation_text']  # Translate classification result
+        classification_result = translation_pipeline(f"أفضل مطابقة: {landmarks_dict[top_label]} باحتمالية {top_prob:.4f}", src_lang='eng_Latn', tgt_lang='arb_Arab')[0]['translation_text']  # Translate classification result
     else:
         caption_translated = caption  # Keep caption in English
         classification_result = f"Best match: {top_label} with probability {top_prob:.4f}"  # Create English classification result
@@ -105,7 +105,7 @@ def process_image(image, language='en'):
         summarized_description = summarize_description(full_description, language)  # Call the summarization function
 
         # Translate caption and classification result
-        caption_translated, classification_result = translate_results(caption, top_label, top_prob, language)  # Call the translation function
+        caption_translated, classification_result = translate_results(caption, top_label, top_prob, landmarks_dict, language)  # Call the translation function
 
         # Convert the summarized description to speech
         audio_file = text_to_speech(summarized_description, language)  # Convert summary to audio
@@ -160,7 +160,7 @@ arabic_interface = gr.Interface(
 # Merge all interfaces into a tabbed interface
 demo = gr.TabbedInterface(
     [english_interface, arabic_interface],  # List of interfaces to include
-    ["English Interface", "الواجهة العربية"]  # Names of the tabs
+    ["English", "العربية"]  # Names of the tabs
 )
 
 # Launch the interface