Update app.py

app.py

@@ -38,7 +38,7 @@ bad_words = st.text_input("Words You Do Not Want Generated", " core lemon height
 
 def run_generate(text, bad_words):
   yo = []
-  input_ids = tokenizer.encode(text, return_tensors='pt').to(device)
+  input_ids = tokenizer.encode(text, return_tensors='pt')
   res = len(tokenizer.encode(text))
   bad_words = bad_words.split()
   bad_word_ids = []