Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -29,9 +29,7 @@ logger = get_logger(__name__)
|
|
29 |
|
30 |
# βββββββ embeddings βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
31 |
# Switch to local BGE embeddings (no JSONDecode errors, no HTTPβbatch issues) :contentReference[oaicite:0]{index=0}
|
32 |
-
|
33 |
-
|
34 |
-
embeddings = SentenceTransformerEmbeddings(
|
35 |
model_name="BAAI/bge-large-en-v1.5",
|
36 |
model_kwargs={"device": "cpu"},
|
37 |
encode_kwargs={"normalize_embeddings": True}
|
@@ -51,7 +49,7 @@ memory = ConversationBufferMemory(
|
|
51 |
)
|
52 |
|
53 |
# βββββββ LLM setup ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
54 |
-
model = "
|
55 |
temperature = 0.1
|
56 |
max_tokens = 500
|
57 |
|
|
|
29 |
|
30 |
# βββββββ embeddings βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
31 |
# Switch to local BGE embeddings (no JSONDecode errors, no HTTPβbatch issues) :contentReference[oaicite:0]{index=0}
|
32 |
+
embeddings = HuggingFaceBgeEmbeddings(
|
|
|
|
|
33 |
model_name="BAAI/bge-large-en-v1.5",
|
34 |
model_kwargs={"device": "cpu"},
|
35 |
encode_kwargs={"normalize_embeddings": True}
|
|
|
49 |
)
|
50 |
|
51 |
# βββββββ LLM setup ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
52 |
+
model = "HuggingFaceTB/SmolLM3-3B"
|
53 |
temperature = 0.1
|
54 |
max_tokens = 500
|
55 |
|