ReithBjarkan commited on
Commit
92a2e60
·
1 Parent(s): 68a368d

Model debugging.

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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
- if model_name.startswith("sentence-transformers"):
33
- # Load model
 
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)")