Update app.py

app.py

@@ -9,6 +9,14 @@ from langchain.prompts import PromptTemplate
 from langchain.llms import HuggingFaceHub
 from langchain.chains import LLMChain
 
+#import model class and tokenizer
+from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+
+#import model class and tokenizer
+from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+
+
+
 ####
 ## Sentinent models
 # https://huggingface.co/spaces/CK42/sentiment-model-comparison
@@ -26,12 +34,17 @@ model_id_6 = "sbcBI/sentiment_analysis_model"
 # https://huggingface.co/spaces/CK42/sentiment-model-comparison
 # 1 seem best for testing
 ####
-#chat_model_facebook-blenderbot_400M_distill = "facebook/blenderbot-400M-distill"
-#chat_model_HenryJJ_vincua_13b = "HenryJJ/vincua-13b"
+#download and setup the model and tokenizer
+model_name = 'facebook/blenderbot-400M-distill'
+tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
+model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
+
+chat_model_facebook_blenderbot_400M_distill = "facebook/blenderbot-400M-distill"
+chat_model_HenryJJ_vincua_13b = "HenryJJ/vincua-13b"
 
 # https://colab.research.google.com/drive/1hrS6_g14EcOD4ezwSGlGX2zxJegX5uNX#scrollTo=NUwUR9U7qkld
 llm_hf = HuggingFaceHub(
-    repo_id= "facebook/blenderbot-400M-distill",
+    repo_id= model,
     model_kwargs={"temperature":0.9 }
 )