jorgik1 commited on
Commit
00ec0e6
·
1 Parent(s): 5694576

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +1 -1
chatbot.py CHANGED
@@ -30,7 +30,7 @@ class YouTubeChatbot:
30
  try:
31
  repo_id = "google/flan-t5-xxl"
32
  self.falcon_llm = HuggingFaceHub(
33
- repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_new_tokens": 1500}
34
  )
35
 
36
  except Exception as e:
 
30
  try:
31
  repo_id = "google/flan-t5-xxl"
32
  self.falcon_llm = HuggingFaceHub(
33
+ repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_new_tokens": 1000}
34
  )
35
 
36
  except Exception as e: