Commit
·
92a2e60
1
Parent(s):
68a368d
Model debugging.
Browse files
app.py
CHANGED
@@ -29,8 +29,9 @@ if st.button("Calculate Similarities"):
|
|
29 |
# Process keywords: Split by commas and/or new lines
|
30 |
keyword_list = [kw.strip() for kw in keywords.replace(",", "\n").split("\n") if kw.strip()]
|
31 |
|
32 |
-
|
33 |
-
|
|
|
34 |
st.info(f"Loading model: {model_name}")
|
35 |
model = SentenceTransformer(model_name)
|
36 |
|
@@ -89,4 +90,4 @@ if st.button("Calculate Similarities"):
|
|
89 |
|
90 |
# Footer
|
91 |
st.markdown("---")
|
92 |
-
st.markdown("Created by Ryland Bacorn(https://huggingface.co/ReithBjarkan)")
|
|
|
29 |
# Process keywords: Split by commas and/or new lines
|
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)
|
37 |
|
|
|
90 |
|
91 |
# Footer
|
92 |
st.markdown("---")
|
93 |
+
st.markdown("Created by [Ryland Bacorn](https://huggingface.co/ReithBjarkan)")
|