Update app.py

app.py

@@ -9,18 +9,6 @@ def calculate_similarity(sentence1, sentence2):
     similarity_score = wl.similarity(sentence1, sentence2)
     return similarity_score
 
-# Define Gradio interface
-iface = gr.Interface(
-    fn=calculate_similarity,
-    inputs=[
-        gr.Textbox(lines=2, placeholder="Enter first sentence..."),
-        gr.Textbox(lines=2, placeholder="Enter second sentence...")
-    ],
-    outputs="number",
-    title="Sentence Similarity with WordLlama",
-    description="Calculate the similarity between two sentences using the WordLlama model from Hugging Face."
-)
-
 # Define five example inputs
 examples = [
     ["I went to the car", "I went to the pawn shop"],
@@ -30,4 +18,20 @@ examples = [
     ["I bought a new phone", "I got a new mobile"]
 ]
 
-iface.launch(share=True, examples=examples)
+# Define Gradio interface
+with gr.Blocks() as iface:
+    gr.Markdown("# Sentence Similarity with WordLlama")
+    gr.Markdown("Calculate the similarity between two sentences using the WordLlama model from Hugging Face.")
+    
+    sentence1 = gr.Textbox(lines=2, placeholder="Enter first sentence...")
+    sentence2 = gr.Textbox(lines=2, placeholder="Enter second sentence...")
+    output = gr.Number()
+    
+    # Button to trigger similarity calculation
+    button = gr.Button("Calculate Similarity")
+    button.click(calculate_similarity, inputs=[sentence1, sentence2], outputs=output)
+    
+    # Examples section
+    gr.Examples(examples=examples, inputs=[sentence1, sentence2])
+
+iface.launch(share=True)