Update app.py

app.py

@@ -7,9 +7,26 @@ class SentenceSimiliarity():
     def __init__(self, sentence1, sentence2):
         self.sentence1 = sentence1
         self.sentence2 = sentence2
-        self.model_name = "bert-base-uncased"
+        
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
         self.model = AutoModel.from_pretrained(self.model_name)
+    
+    def model_selection(self):
+        available_models = [
+            "distilbert-base-uncased",
+            "bert-base-uncased",
+            "sentence-transformers/all-MiniLM-L6-v2",
+            "sentence-transformers/all-mpnet-base-v2",
+            "intfloat/multilingual-e5-base",
+            "togethercomputer/m2-bert-80M-32k-retrieval",
+            "togethercomputer/m2-bert-80M-8k-retrieval",
+            "togethercomputer/m2-bert-80M-2k-retrieval",
+        ]
+        self.model_name = st.sidebar.selectbox(
+            label="Select Your Models",
+            options=available_models,
+        )
+
 
     def tokenize(self):
         tokenized1 = self.tokenizer(