Hugging Face's logo

Spaces:
jorgik1
/
youtuber-ai-chatbot
Sleeping

App Files Community
jorgik1 commited on
Commit
39ebf86
·
1 Parent(s): 1f46c5a

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +1 -1
chatbot.py CHANGED
@@ -28,7 +28,7 @@ class YouTubeChatbot:
28
  self.embeddings = None
29
 
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
  )
 
28
  self.embeddings = None
29
 
30
  try:
31
+ repo_id = "PygmalionAI/pygmalion-6b"
32
  self.falcon_llm = HuggingFaceHub(
33
  repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_new_tokens": 1000}
34
  )