Update app.py

app.py

@@ -21,8 +21,8 @@ def get_pipeline():
         def preprocess(self, inputs):
             question, essay = inputs
             encoding = self.tokenizer(question, essay, return_tensors='pt', padding='max_length', truncation=True, max_length=512)
-            input_ids = encoding['input_ids'].to(device)
-            attention_mask = encoding['attention_mask'].to(device)
+            input_ids = encoding['input_ids']
+            attention_mask = encoding['attention_mask']
             return {'input_ids': input_ids, 'attention_mask': attention_mask}
         
         def _forward(self, input):