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

Model debugging.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)