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)
|