added debug prints
Browse files
app.py
CHANGED
@@ -47,7 +47,9 @@ def process(path, progress = gr.Progress()):
|
|
47 |
generated_ids = model.generate(torch.cat(batch))
|
48 |
progress(0, desc="Decoding (token -> str)")
|
49 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
|
|
50 |
full_text = " ".join(generated_text)
|
|
|
51 |
|
52 |
progress(0, desc="Correction..")
|
53 |
fix_spelling = pipeline("text2text-generation",model=CORRECTOR_PATH)
|
|
|
47 |
generated_ids = model.generate(torch.cat(batch))
|
48 |
progress(0, desc="Decoding (token -> str)")
|
49 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
50 |
+
print(generated_text)
|
51 |
full_text = " ".join(generated_text)
|
52 |
+
print(full_text)
|
53 |
|
54 |
progress(0, desc="Correction..")
|
55 |
fix_spelling = pipeline("text2text-generation",model=CORRECTOR_PATH)
|