Commit
·
d14b366
1
Parent(s):
92a2e60
Model debugging.
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ st.text_area(
|
|
18 |
key="keywords"
|
19 |
)
|
20 |
keywords = st.session_state.keywords
|
21 |
-
model_name = st.selectbox("Select Embedding Model", ["sentence-transformers/LaBSE", "all-MiniLM-L6-v2", "OpenAI Embeddings"])
|
22 |
openai_api_key = st.text_input("OpenAI API Key (optional)", type="password")
|
23 |
|
24 |
# Process Button
|
@@ -30,7 +30,7 @@ if st.button("Calculate Similarities"):
|
|
30 |
keyword_list = [kw.strip() for kw in keywords.replace(",", "\n").split("\n") if kw.strip()]
|
31 |
|
32 |
# Check model type
|
33 |
-
if model_name in ["sentence-transformers/LaBSE", "all-MiniLM-L6-v2"]:
|
34 |
# Load SentenceTransformer model
|
35 |
st.info(f"Loading model: {model_name}")
|
36 |
model = SentenceTransformer(model_name)
|
|
|
18 |
key="keywords"
|
19 |
)
|
20 |
keywords = st.session_state.keywords
|
21 |
+
model_name = st.selectbox("Select Embedding Model", ["sentence-transformers/LaBSE", "sentence-transformers/all-MiniLM-L6-v2", "OpenAI Embeddings"])
|
22 |
openai_api_key = st.text_input("OpenAI API Key (optional)", type="password")
|
23 |
|
24 |
# Process Button
|
|
|
30 |
keyword_list = [kw.strip() for kw in keywords.replace(",", "\n").split("\n") if kw.strip()]
|
31 |
|
32 |
# Check model type
|
33 |
+
if model_name in ["sentence-transformers/LaBSE", "sentence-transformers/all-MiniLM-L6-v2"]:
|
34 |
# Load SentenceTransformer model
|
35 |
st.info(f"Loading model: {model_name}")
|
36 |
model = SentenceTransformer(model_name)
|