Spaces:

sabahat-shakeel
/

chat-bot

Sleeping

sabahat-shakeel commited on Jan 29

Commit

1f5c69e

verified ·

1 Parent(s): af7d593

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,7 +47,7 @@ st.title("🤖 Smart Chatbot")
 @st.cache_resource
 def load_model():
-    model_name = "facebook/blenderbot-400M-distill"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     return pipeline("text2text-generation", model=model, tokenizer=tokenizer)

 @st.cache_resource
 def load_model():
+    model_name =  "facebook/blenderbot-3B"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     return pipeline("text2text-generation", model=model, tokenizer=tokenizer)