Update app.py

app.py

@@ -29,10 +29,10 @@ load_dataset("Ghosthash/Tweets")
 
 from transformers import pipeline
 
-pipe = pipeline("text-generation", model="cognitivecomputations/dolphin-2.8-mistral-7b-v02", max_new_tokens=1000)
+pipe = pipeline("text-generation", model="cognitivecomputations/dolphin-2.8-mistral-7b-v02", device=1)
 text = st.text_input("enter text here")
 
 if text:
-    response = pipe(text)
+    response = pipe(text, max_new_tokens=1000)
     st.json(response)