Update app.py

app.py

@@ -182,7 +182,9 @@ def ocr_correction(prompt, max_new_tokens=600):
     # Decode and return the generated text
     result = tokenizer.decode(output[0], skip_special_tokens=True)
 
-    result = result.split("### Correction ###\n")[1]
+    print(result)
+
+    result = result.split("### Correction ###")[1]
 
     return result