jacobmp commited on
Commit
c9bae28
·
verified ·
1 Parent(s): 939af4a

added debug prints

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)