Update app.py

app.py

@@ -9,8 +9,8 @@ def text_processing(text):
     inputs = [text]
 
     # Tokenize and prepare the inputs for model
-    input_ids = tokenizer(inputs, return_tensors="pt", max_length=512, truncation=True, padding="max_length").input_ids.to(device)
-    attention_mask = tokenizer(inputs, return_tensors="pt", max_length=512, truncation=True, padding="max_length").attention_mask.to(device)
+    input_ids = tokenizer(inputs, return_tensors="pt", max_length=512, truncation=True, padding="max_length").input_ids
+    attention_mask = tokenizer(inputs, return_tensors="pt", max_length=512, truncation=True, padding="max_length").attention_mask
 
     # Generate prediction
     output = model.generate(input_ids=input_ids, attention_mask=attention_mask, max_new_tokens=512)