nurqoneah commited on
Commit
79f4297
·
verified ·
1 Parent(s): d3ef806

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/SeaLLM-7B-v2.5-mlx-quantized"
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
- # return llm
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)