Updated for CPU usage

app.py

@@ -8,7 +8,8 @@ model_name = "microsoft/DialoGPT-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 model = AutoModelForCausalLM.from_pretrained(model_name)
-chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0, truncation=True, max_length=1000)
+chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=-1, truncation=True, max_length=1000)
+
 
 # Streamlit UI
 st.title("Buddy Christ Chatbot 🌟")

requirements.txt

@@ -1,4 +1,4 @@
-torch
-torchvision
+torch==1.9.0+cpu
+torchvision==0.10.0+cpu
 transformers
 streamlit