innocent-charles commited on
Commit
8e1de57
1 Parent(s): 50fdd96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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
- model_dropdown = gr.Dropdown(choices=list(model_dict.keys()), label="Model")
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 = [model_dropdown, mode_dropdown, sentence1_input, sentence2_input, sentence3_input, sentence4_input, dimension_dropdown]
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 = [