fix: batching logic

app.py

@@ -52,11 +52,10 @@ def process_records_gradio(records, fields, question, example_records=None):
         task.set_runtime_parameters(runtime_parameters)
 
         results = []
-        print(records[0])
         output = task.process(inputs=[{"record": record} for record in records])
-
-        for _ in records:
-            entry = next(output)[0]
+        output = next(output)
+        for idx in range(len(records)):
+            entry = output[idx]
             if entry["suggestions"]:
                 results.append(entry["suggestions"])