Update app.py

app.py

@@ -172,12 +172,8 @@ def ocr_correction(prompt, max_new_tokens=500):
     prompt = f"""### Text ###\n{prompt}\n\n\n### Correction ###\n"""
     input_ids = tokenizer.encode(prompt, return_tensors="pt").to(device)
 
-    # Set the number of threads for PyTorch
-    torch.set_num_threads(num_threads)
-
     # Generate text
-    output = model.generate,
-            input_ids,
+    output = model.generate(input_ids,
             max_new_tokens=max_new_tokens,
             pad_token_id=tokenizer.eos_token_id,
             top_k=50,