Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -82,7 +82,7 @@ def get_response(user_input):
|
|
| 82 |
# lib="avx2", # for CPU
|
| 83 |
# )
|
| 84 |
|
| 85 |
-
llm_model == "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
| 86 |
llm = HuggingFaceHub(
|
| 87 |
repo_id=llm_model,
|
| 88 |
model_kwargs={"temperature": temperature, "max_new_tokens": 250, "top_k": top_k}
|
|
|
|
| 82 |
# lib="avx2", # for CPU
|
| 83 |
# )
|
| 84 |
|
| 85 |
+
llm_model == "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
| 86 |
llm = HuggingFaceHub(
|
| 87 |
repo_id=llm_model,
|
| 88 |
model_kwargs={"temperature": temperature, "max_new_tokens": 250, "top_k": top_k}
|