innocent-charles
commited on
Commit
•
8e1de57
1
Parent(s):
50fdd96
Update app.py
Browse files
app.py
CHANGED
@@ -69,15 +69,13 @@ def predict(model_name, mode, sentence1, sentence2=None, sentence3=None, sentenc
|
|
69 |
return result
|
70 |
|
71 |
# Define inputs and outputs for Gradio interface
|
72 |
-
|
73 |
-
mode_dropdown = gr.Dropdown(choices=["Compare two sentences", "Compare one to three"], label="Mode")
|
74 |
-
dimension_dropdown = gr.Dropdown(choices=["768", "512", "256", "128", "64"], label="Embedding Dimension")
|
75 |
sentence1_input = gr.Textbox(lines=2, placeholder="Enter the first sentence here...", label="Sentence 1")
|
76 |
sentence2_input = gr.Textbox(lines=2, placeholder="Enter the second sentence here...", label="Sentence 2 (or first of three for mode)")
|
77 |
sentence3_input = gr.Textbox(lines=2, placeholder="Enter the third sentence here...", label="Sentence 3")
|
78 |
sentence4_input = gr.Textbox(lines=2, placeholder="Enter the fourth sentence here...", label="Sentence 4")
|
79 |
|
80 |
-
inputs = [
|
81 |
outputs = gr.JSON(label="Detailed Similarity Scores")
|
82 |
|
83 |
examples = [
|
|
|
69 |
return result
|
70 |
|
71 |
# Define inputs and outputs for Gradio interface
|
72 |
+
model_name = "AviLABASE", label="Model")
|
|
|
|
|
73 |
sentence1_input = gr.Textbox(lines=2, placeholder="Enter the first sentence here...", label="Sentence 1")
|
74 |
sentence2_input = gr.Textbox(lines=2, placeholder="Enter the second sentence here...", label="Sentence 2 (or first of three for mode)")
|
75 |
sentence3_input = gr.Textbox(lines=2, placeholder="Enter the third sentence here...", label="Sentence 3")
|
76 |
sentence4_input = gr.Textbox(lines=2, placeholder="Enter the fourth sentence here...", label="Sentence 4")
|
77 |
|
78 |
+
inputs = [model_name, sentence1_input, sentence2_input, sentence3_input, sentence4_input]
|
79 |
outputs = gr.JSON(label="Detailed Similarity Scores")
|
80 |
|
81 |
examples = [
|