Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -263,8 +263,8 @@ def build_qa_chain(collection_name, vector_db, file: str):
|
|
| 263 |
|
| 264 |
"""
|
| 265 |
#vincent for new LLM
|
| 266 |
-
|
| 267 |
-
llm_model = "meta-llama/Llama-2-7b-chat-hf"
|
| 268 |
llm = HuggingFaceEndpoint(
|
| 269 |
repo_id=llm_model,
|
| 270 |
task="text-generation", # Explicitly specify task
|
|
|
|
| 263 |
|
| 264 |
"""
|
| 265 |
#vincent for new LLM
|
| 266 |
+
llm_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
| 267 |
+
#llm_model = "meta-llama/Llama-2-7b-chat-hf"
|
| 268 |
llm = HuggingFaceEndpoint(
|
| 269 |
repo_id=llm_model,
|
| 270 |
task="text-generation", # Explicitly specify task
|