Update app.py

app.py

@@ -4,30 +4,12 @@ from transformers import AutoTokenizer, AutoModel
 from sentence_transformers import util
 class SentenceSimiliarity():
 
-    def __init__(self, sentence1, sentence2):
+    def __init__(self, model_name, model, sentence1, sentence2):
         self.sentence1 = sentence1
         self.sentence2 = sentence2
-        
+        self.model_name = model_name
+        self.model = model
         self.tokenizer = AutoTokenizer.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,
-        )
-        self.model = AutoModel.from_pretrained(self.model_name)
-
 
     def tokenize(self):
         tokenized1 = self.tokenizer(
@@ -83,9 +65,26 @@ class UI():
             help='Check Sentence Similarity'
         )
 
+    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,
+        )
+        self.model = AutoModel.from_pretrained(self.model_name)
+
     def result(self):
         self.get()
-        ss = SentenceSimiliarity(self.sentence1, self.sentence2)
+        ss = SentenceSimiliarity(self.model_name, self.model, self.sentence1, self.sentence2)
         
         if self.button:
             st.text(ss.results())