Pclanglais commited on
Commit
fa86caf
1 Parent(s): e103855

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -194,9 +194,7 @@ class OCRCorrector:
194
  self.system_prompt = system_prompt
195
 
196
  def correct(self, user_message):
197
- sampling_params = SamplingParams(temperature=0.9, top_p=0.95, max_tokens=4000, presence_penalty=0, stop=["#END#"])
198
- detailed_prompt = f"### Text ###\n{user_message}\n\n### Correction ###\n"
199
- generated_text = ocr_correction(detailed_prompt)
200
  html_diff = generate_html_diff(user_message, generated_text)
201
  return generated_text, html_diff
202
 
 
194
  self.system_prompt = system_prompt
195
 
196
  def correct(self, user_message):
197
+ generated_text = ocr_correction(user_message)
 
 
198
  html_diff = generate_html_diff(user_message, generated_text)
199
  return generated_text, html_diff
200