Update app.py

app.py

@@ -52,31 +52,31 @@ def load_index(model):
 
 
 dict_models = {
-	'en-es': model_es,
-	'en-en': model_en,
-	'en-de': model_de,
+	'fr-es': model_es,
+	'fr-en': model_en,
+	'fr-de': model_de,
 	# 'en-ar': model_ar,
 }
 
 dict_models_tr = {
-	'en-es': model_tr_es,
-	'en-en': model_tr_en,
-	'en-de': model_tr_de,
+	'fr-es': model_tr_es,
+	'fr-en': model_tr_en,
+	'fr-de': model_tr_de,
 	# 'en-ar': model_tr_ar,
 }
 
 dict_tokenizer_tr = {
-	'en-es': tokenizer_es,
-	'en-en': tokenizer_en,
-	'en-de': tokenizer_de,
+	'fr-es': tokenizer_es,
+	'fr-en': tokenizer_en,
+	'fr-de': tokenizer_de,
 	# 'en-ar': tokenizer_ar,
 }
 # dict_reference_faiss = {'en-es':[]}
 dict_reference_faiss = {
-	'en-es': [],  #load_index('en-es'),
-	'en-ar': [], #load_index('en-ar'),
-	'en-fr': [], #load_index('en-fr'),
-	'en-zh': [], #load_index('en-zh'),
+	'fr-es': [],  #load_index('en-es'),
+	'fr-en': [], #load_index('en-ar'),
+	'fr-de': [], #load_index('en-fr'),
+	'fr-zh': [], #load_index('en-zh'),
 }
 
 # print("dict", dict_reference_faiss['en-es']['input']['tokens'][1])
@@ -103,11 +103,31 @@ contrastive_examples = [
 	]
 
 
+french_examples = [
+    ["","","Les appels répétés de sa mère [auraient] dû nous alerter.", "", ""],
+    ["","","Elle a promis à son médecin de demeurer [active] après s’être retirée.", "", ""],
+    ["","","Nous [avons] lancé une insulte et nous [sommes] partis brusquement.", "",""],
+    ["","","La vache et la poule [doivent] être [nourries].","",""],
+    ["","","Le bruit soudain dans les chambres supérieures [aurait] dû nous alerter.","",""],
+    ["","","le bailleur de fonds a terminé son travail.","",""],
+    ["","","la bailleuse de fonds a terminé son travail.","",""],
+    ["","","l'esthéticien a terminé son travail.","",""],
+    ["","","l'esthéticienne a terminé son travail.","",""],
+    ["","","l'esthéticienne a terminé son travail.","",""],
+    ["","","chaque ingénieure a terminé son travail.","",""],
+    ["","","chaque ingénieur a fini son travail.","",""],
+    
+
+    
+    
+]
+
 
 #Load challenge set examples
-df_challenge_set = pd.read_csv("challenge_sets.csv")
-arr_challenge_set = df_challenge_set.values
-arr_challenge_set = [[x[2], x[3], x[4], x[5]] for x in arr_challenge_set]
+# df_challenge_set = pd.read_csv("challenge_sets.csv")
+# arr_challenge_set = df_challenge_set.values
+# arr_challenge_set = [[x[2], x[3], x[4], x[5]] for x in arr_challenge_set]
+arr_challenge_set = [[x[2]] for x in french_examples]
 
 
 
@@ -828,7 +848,7 @@ with gr.Blocks(js="plotsjs.js") as demo:
 		"""
 		1. Select the language pair for the translation 
 		""")
-	radio_c = gr.Radio(choices=['fr-en', 'fr-es', 'fr-de'], value="fr-en", label= ['French to English', "French to Spanish", "French to German"], container=False)
+	radio_c = gr.Radio(choices=[['french to english','fr-en'], ['french to spanish','fr-es'], ['french to german', 'fr-de']], value="fr-en", label= ['French to English', "French to Spanish", "French to German"], container=False)
 	gr.Markdown(
 		"""
 		2. Source text to translate
@@ -839,12 +859,13 @@ with gr.Blocks(js="plotsjs.js") as demo:
 			"""
 			### select an example from the challenge set listed bellow
 			""")
-		challenge_ex  = gr.Textbox(label="Challenge", interactive=False)
-		category_minor  = gr.Textbox(label="category_minor", interactive=False)
-		category_major  = gr.Textbox(label="category_major", interactive=False)
+		# challenge_ex  = gr.Textbox(label="Challenge", interactive=False)
+		# category_minor  = gr.Textbox(label="category_minor", interactive=False)
+		# category_major  = gr.Textbox(label="category_major", interactive=False)
 		
 		with gr.Accordion("Examples:"):
-			gr.Examples(arr_challenge_set,[in_text, challenge_ex,category_minor,category_major], label="")
+			gr.Examples(arr_challenge_set,[in_text], label="")
+			# gr.Examples(arr_challenge_set,[in_text, challenge_ex,category_minor,category_major], label="")
 
 	btn = gr.Button("Translate")