Update app.py

app.py

@@ -4,7 +4,7 @@ from datasets import load_dataset
 
 load_dataset("kaifahmad/Hate-Speech-Tweets")
 # pipe = pipeline("text-generation", model="cognitivecomputations/dolphin-2.7-mixtral-8x7b")
-pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B")
+pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B", device=1)
 text = st.text_input('enter text here')
 
 if text: