Update app.py

app.py

@@ -35,9 +35,9 @@ batch["reference"] = processor.tokenizer._normalize(batch['text'])
 
 with torch.no_grad():
     predicted_ids = model.generate(input_features.to("cuda"))[0]
-transcription = processor.decode(predicted_ids)
-batch["prediction"] = processor.tokenizer._normalize(transcription)
-return batch
+    transcription = processor.decode(predicted_ids)
+    batch["prediction"] = processor.tokenizer._normalize(transcription)
+    return batch
 
 result = dataset.map(map_to_pred)