adding some texts

app.py

@@ -107,16 +107,24 @@ try:
 
     # set the model to eval mode
     _ = model.eval()
-
+    
+    language = "Wolof" if translation_type == "French ➡️ Wolof" else "French"
+    
     # Add a title
-    st.header("Translate French sentences onto Wolof 👌")
-
-
+    st.header(f"Translate French sentences to {language} 👌")
+    
     # Recuperate two columns
     left, right = st.columns(2)
 
-    # recuperate sentences
-    left.subheader('Give me some sentences in French: ')
+    if translation_type == "French ➡️ Wolof":
+
+        # recuperate sentences
+        left.subheader('Give me some sentences in French: ')
+    
+    else:
+
+        # recuperate sentences
+        left.subheader('Give me some sentences in Wolof: ')
 
     for i in range(number):
         
@@ -125,8 +133,15 @@ try:
     # run model inference on all test data
     original_translations, predicted_translations, original_texts, scores = [], [], [], {}
 
-    # print a sentence recuperated from the session
-    right.subheader("Translation to Wolof:")
+    if translation_type == "French ➡️ Wolof":
+        
+        # print a sentence recuperated from the session
+        right.subheader("Translation to Wolof:")
+    
+    else:
+        
+        # print a sentence recuperated from the session
+        right.subheader("Translation to French:")
 
     for i in range(number):