Frenchizer commited on
Commit
65f8dcc
·
verified ·
1 Parent(s): e7dcd52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from functools import lru_cache
10
 
11
  @lru_cache(maxsize=1)
12
  def load_model_and_tokenizer():
13
- model_name = "Frenchizer/all-MiniLM-L6-v2" # Replace with your Space and model path
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
15
  model = AutoModel.from_pretrained(model_name)
16
  return tokenizer, model
 
10
 
11
  @lru_cache(maxsize=1)
12
  def load_model_and_tokenizer():
13
+ model_name = "./all-MiniLM-L6-v2"
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
15
  model = AutoModel.from_pretrained(model_name)
16
  return tokenizer, model