mindspark121 commited on
Commit
45eb46b
·
verified ·
1 Parent(s): e1d5461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -21,8 +21,7 @@ if not OPENROUTER_API_KEY:
21
  raise ValueError("❌ OPENROUTER_API_KEY is missing. Set it as an environment variable.")
22
 
23
  OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1/chat/completions"
24
-
25
- embedding_model = SentenceTransformer("sentence-transformers/paraphrase-MiniLM-L6-v2", cache_folder="/app/cache")
26
  summarization_model = AutoModelForSeq2SeqLM.from_pretrained("google/long-t5-tglobal-base", cache_dir="/app/cache")
27
  summarization_tokenizer = AutoTokenizer.from_pretrained("google/long-t5-tglobal-base", cache_dir="/app/cache")
28
 
 
21
  raise ValueError("❌ OPENROUTER_API_KEY is missing. Set it as an environment variable.")
22
 
23
  OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1/chat/completions"
24
+ embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder="/app/cache")
 
25
  summarization_model = AutoModelForSeq2SeqLM.from_pretrained("google/long-t5-tglobal-base", cache_dir="/app/cache")
26
  summarization_tokenizer = AutoTokenizer.from_pretrained("google/long-t5-tglobal-base", cache_dir="/app/cache")
27