euler314 commited on
Commit
592abec
·
verified ·
1 Parent(s): 7a71728

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ def load_models():
102
  # Load SentenceTransformer with offline handling
103
  try:
104
  from sentence_transformers import SentenceTransformer
105
- model_name = 'deepseek-ai/DeepSeek-R1-Distill-Qwen-7B'
106
  cache_dir = os.path.expanduser('~/.cache/torch/sentence_transformers')
107
  if os.path.exists(os.path.join(cache_dir, model_name)):
108
  semantic_model = SentenceTransformer(os.path.join(cache_dir, model_name))
 
102
  # Load SentenceTransformer with offline handling
103
  try:
104
  from sentence_transformers import SentenceTransformer
105
+ model_name = 'deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B'
106
  cache_dir = os.path.expanduser('~/.cache/torch/sentence_transformers')
107
  if os.path.exists(os.path.join(cache_dir, model_name)):
108
  semantic_model = SentenceTransformer(os.path.join(cache_dir, model_name))