captain-awesome commited on
Commit
94701f4
1 Parent(s): a9a1953

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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}