Update app.py

app.py

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