Spaces:
Sleeping
Sleeping
Sharathhebbar24
commited on
Commit
•
8b70035
1
Parent(s):
c9db1ed
Update app.py
Browse files
app.py
CHANGED
@@ -7,9 +7,26 @@ class SentenceSimiliarity():
|
|
7 |
def __init__(self, sentence1, sentence2):
|
8 |
self.sentence1 = sentence1
|
9 |
self.sentence2 = sentence2
|
10 |
-
|
11 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
12 |
self.model = AutoModel.from_pretrained(self.model_name)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
|
14 |
def tokenize(self):
|
15 |
tokenized1 = self.tokenizer(
|
|
|
7 |
def __init__(self, sentence1, sentence2):
|
8 |
self.sentence1 = sentence1
|
9 |
self.sentence2 = sentence2
|
10 |
+
|
11 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
12 |
self.model = AutoModel.from_pretrained(self.model_name)
|
13 |
+
|
14 |
+
def model_selection(self):
|
15 |
+
available_models = [
|
16 |
+
"distilbert-base-uncased",
|
17 |
+
"bert-base-uncased",
|
18 |
+
"sentence-transformers/all-MiniLM-L6-v2",
|
19 |
+
"sentence-transformers/all-mpnet-base-v2",
|
20 |
+
"intfloat/multilingual-e5-base",
|
21 |
+
"togethercomputer/m2-bert-80M-32k-retrieval",
|
22 |
+
"togethercomputer/m2-bert-80M-8k-retrieval",
|
23 |
+
"togethercomputer/m2-bert-80M-2k-retrieval",
|
24 |
+
]
|
25 |
+
self.model_name = st.sidebar.selectbox(
|
26 |
+
label="Select Your Models",
|
27 |
+
options=available_models,
|
28 |
+
)
|
29 |
+
|
30 |
|
31 |
def tokenize(self):
|
32 |
tokenized1 = self.tokenizer(
|