Update app.py

app.py

@@ -163,7 +163,7 @@ def split_text(text, max_tokens=500):
 
 
 # Function to generate text
- def ocr_correction(prompt, max_new_tokens=600, num_threads=os.cpu_count()):
+def ocr_correction(prompt, max_new_tokens=600, num_threads=os.cpu_count()):
     prompt = f"""### Text ###\n{prompt}\n\n\n### Correction ###\n"""
     input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)