Update app.py

app.py

@@ -8,7 +8,7 @@ class SentenceSimiliarity():
         self.sentence1 = sentence1
         self.sentence2 = sentence2
         self.model_name = model_name
-        self.model = model
+        self.model = AutoModel.from_pretrained(self.model_name)
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
 
     def tokenize(self):
@@ -76,15 +76,17 @@ class UI():
             "togethercomputer/m2-bert-80M-8k-retrieval",
             "togethercomputer/m2-bert-80M-2k-retrieval",
         ]
-        self.model_name = st.sidebar.selectbox(
+        model_name = st.sidebar.selectbox(
             label="Select Your Models",
             options=available_models,
         )
-        self.model = AutoModel.from_pretrained(self.model_name)
+        return model_name
+        
 
     def result(self):
         self.get()
-        ss = SentenceSimiliarity(self.model_name, self.model, self.sentence1, self.sentence2)
+        model_name = self.model_selection()
+        ss = SentenceSimiliarity(model_name, self.sentence1, self.sentence2)
         
         if self.button:
             st.text(ss.results())