Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModel
 from sentence_transformers import util
 class SentenceSimiliarity():
 
-    def __init__(self, model_name, model, sentence1, sentence2):
+    def __init__(self, model_name, sentence1, sentence2):
         self.sentence1 = sentence1
         self.sentence2 = sentence2
         self.model_name = model_name
@@ -70,11 +70,11 @@ class UI():
             "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",
+            # "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",
         ]
         model_name = st.sidebar.selectbox(
             label="Select Your Models",