stefanoviel commited on
Commit
ca806f9
·
1 Parent(s): c25929c
Files changed (1) hide show
  1. src/streamlit_app.py +4 -1
src/streamlit_app.py CHANGED
@@ -1,12 +1,15 @@
1
  import os
 
2
  import streamlit as st
3
  import pandas as pd
4
  from sentence_transformers import SentenceTransformer, util
5
  import torch
6
  from spellchecker import SpellChecker
7
  from io import StringIO
8
- from pathlib import Path
9
 
 
 
 
10
  script_dir = Path(__file__).parent
11
 
12
  # Define paths relative to the script's directory
 
1
  import os
2
+ from pathlib import Path
3
  import streamlit as st
4
  import pandas as pd
5
  from sentence_transformers import SentenceTransformer, util
6
  import torch
7
  from spellchecker import SpellChecker
8
  from io import StringIO
 
9
 
10
+ # --- Configuration ---
11
+ EMBEDDING_MODEL = 'sentence-transformers/all-MiniLM-L6-v2'
12
+ # Use /tmp directory for temporary files in Hugging Face Spaces
13
  script_dir = Path(__file__).parent
14
 
15
  # Define paths relative to the script's directory