Spaces:
Running
Running
Pclanglais
commited on
Commit
•
63e8ceb
1
Parent(s):
10a4171
Update app.py
Browse files
app.py
CHANGED
@@ -141,7 +141,7 @@ def ocr_correction(prompt, max_new_tokens=600):
|
|
141 |
results = generator.generate_batch(
|
142 |
list_prompts,
|
143 |
max_length=max_new_tokens,
|
144 |
-
sampling_temperature=0
|
145 |
sampling_topk=20,
|
146 |
repetition_penalty=1.1,
|
147 |
include_prompt_in_result=False
|
|
|
141 |
results = generator.generate_batch(
|
142 |
list_prompts,
|
143 |
max_length=max_new_tokens,
|
144 |
+
sampling_temperature=0,
|
145 |
sampling_topk=20,
|
146 |
repetition_penalty=1.1,
|
147 |
include_prompt_in_result=False
|