Update app.py

app.py

@@ -96,8 +96,8 @@ def generate_output(test_samples):
         max_length=1024,
         return_tensors="pt",
     )
-    input_ids = inputs.input_ids.to(model.)
-    attention_mask = inputs.attention_mask.to(model)
+    input_ids = inputs.input_ids
+    attention_mask = inputs.attention_mask
     outputs = model.generate(input_ids,
                              max_length = 256,
                              min_length=32,