Update app.py

app.py

@@ -7,7 +7,7 @@ def load_model(model_name):
     return SentenceTransformer(model_name)
 
 # Function to compute similarity and classify relationship
-def predict(model_name_display, original_sentence_input, sentence_1=None, sentence_2=None, sentence_3=None):
+def predict(model_name_display, original_sentence_input, sentence_1=None, sentence_2=None, sentence_3=None, sentence_4=None, sentence_5=None):
     model_name = "sartifyllc/African-Cross-Lingua-Embeddings-Model"
     model = load_model(model_name)
     result = {
@@ -16,7 +16,9 @@ def predict(model_name_display, original_sentence_input, sentence_1=None, senten
         "Sentences to Compare": {
             "Sentence 1": sentence_1,
             "Sentence 2": sentence_2,
-            "Sentence 3": sentence_3
+            "Sentence 3": sentence_3,
+            "Sentence 4": sentence_4,
+            "Sentence 5": sentence_5
         },
         "Similarity Scores": {}
     }
@@ -32,7 +34,9 @@ def predict(model_name_display, original_sentence_input, sentence_1=None, senten
     similarity_scores = {
         "Sentence 1": float(similarities[0, 0]),
         "Sentence 2": float(similarities[0, 1]),
-        "Sentence 3": float(similarities[0, 2])
+        "Sentence 3": float(similarities[0, 2]),
+        "Sentence 4": float(similarities[0, 3]),
+         "Sentence 5": float(similarities[0, 4]),
     }
     result["Similarity Scores"] = similarity_scores
     
@@ -41,9 +45,12 @@ def predict(model_name_display, original_sentence_input, sentence_1=None, senten
 model_name_display = gr.Markdown(value="**Model Name**: sartifyllc/African-Cross-Lingua-Embeddings-Model")
 original_sentence_input = gr.Textbox(lines=2, placeholder="Enter the original sentence here...", label="Original Sentence")
 
-sentence_1 = gr.Textbox(lines=2, placeholder="Enter the first sentence here...", label="Sentence 1")
-sentence_2 = gr.Textbox(lines=2, placeholder="Enter the second sentence here...", label="Sentence 2")
-sentence_3 = gr.Textbox(lines=2, placeholder="Enter the third sentence here...", label="Sentence 3")
+sentence_1 = gr.Textbox(lines=2, placeholder="Enter the sentence to compare here...", label="Sentence 1")
+sentence_2 = gr.Textbox(lines=2, placeholder="Enter the sentence to compare here...", label="Sentence 2")
+sentence_3 = gr.Textbox(lines=2, placeholder="Enter the sentence to compare here...", label="Sentence 3")
+sentence_4 = gr.Textbox(lines=2, placeholder="Enter the sentence to compare here...", label="Sentence 4")
+sentence_5 = gr.Textbox(lines=2, placeholder="Enter the sentence to compare here...", label="Sentence 5")
+
 
 inputs = [model_name_display, original_sentence_input, sentence_1, sentence_2, sentence_3]
 outputs = gr.JSON(label="Detailed Similarity Scores")
@@ -63,21 +70,27 @@ gr.Interface(
         "Jua linawaka sana leo.",
         "Òrùlé jẹ́ tí ó ti máa ń tan-ìmólẹ̀ lónìí.",
         "Ran na haske sosai yau.",
-        "The sun is shining brightly today."
+        "The sun is shining brightly today.",
+        "napenda sana jua",
+        "schooling is kinda boring, I don't like it"
     ],
     [
         "sartifyllc/African-Cross-Lingua-Embeddings-Model",
         "Mbuzi anaruka juu ya uzio.",
         "Àgbò ohun ti ń kọjú wá sórí orí-ọ̀kẹ́.",
         "Kura mai sauri tana tsalle kan kare mai barci.",
-        "The goat is jumping over the fence."
+        "The goat is jumping over the fence.",
+        "Àgbò ohun ti ń kọjú wá sórí orí-ọ̀kẹ́.",
+        "The cat is sleeping under the table."
     ],
     [
         "sartifyllc/African-Cross-Lingua-Embeddings-Model",
         "Ninapenda kujifunza lugha mpya.",
         "Mo nífẹ̀ẹ́ láti kọ́ èdè tuntun.",
         "Ina son koyon sababbin harsuna.",
-        "I love learning new languages."
+        "I love learning new languages.",
+        "Ina son koyon sababbin harsuna.",
+        "botu neyle ki lo no"
     ]
 ]