Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ INITIAL_MESSAGE = """Halo! 👋 Saya adalah asisten kesehatan feminacare yang si
|
|
24 |
Silakan ajukan pertanyaan apa saja dan saya akan membantu Anda dengan informasi yang akurat."""
|
25 |
|
26 |
# Model configurations
|
27 |
-
MODEL_NAME = "SeaLLMs/
|
28 |
EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
|
29 |
TOP_K_DOCS = 5
|
30 |
|
@@ -85,7 +85,7 @@ def create_llm():
|
|
85 |
repetition_penalty=1.03,
|
86 |
),
|
87 |
)
|
88 |
-
|
89 |
|
90 |
|
91 |
# chat_model = ChatHuggingFace(llm=llm)
|
|
|
24 |
Silakan ajukan pertanyaan apa saja dan saya akan membantu Anda dengan informasi yang akurat."""
|
25 |
|
26 |
# Model configurations
|
27 |
+
MODEL_NAME = "SeaLLMs/SeaLLMs-v3-7B-Chat"
|
28 |
EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
|
29 |
TOP_K_DOCS = 5
|
30 |
|
|
|
85 |
repetition_penalty=1.03,
|
86 |
),
|
87 |
)
|
88 |
+
return llm
|
89 |
|
90 |
|
91 |
# chat_model = ChatHuggingFace(llm=llm)
|