Updated the chatbot to use DialoGPT-small model

app.py

@@ -1,13 +1,14 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline 
 
-# Initialize the model and tokenizer
-model_name = "TheBloke/Mistral-7B-OpenOrca-GGUF"
-base_model_name = "EleutherAI/gpt-j-1.1B"  # Base model for tokenizer
-tokenizer = AutoTokenizer.from_pretrained("TheBloke/Mistral-7B-OpenOrca-GGUF")
+# Use a smaller model 
+model_name = "microsoft/DialoGPT-small"  
+
+# Use the model's default GPT2 tokenizer
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 model = AutoModelForCausalLM.from_pretrained(model_name)
-chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)  # device=0 means use CPU
+chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0, truncation=True, max_length=1000)
 
 # Streamlit UI
 st.title("Buddy Christ Chatbot 🌟")