Update app.py

app.py

@@ -71,10 +71,15 @@ def process_pdf_to_epub(pdf_file, title, author):
                 num_return_sequences=1,
                 do_sample=True,
             )
-
             prompt_length = inputs["input_ids"].shape[1]
             new_tokens = output[:, prompt_length:]
-            decoded = processor.tokenizer.batch_decode(new_tokens, skip_special_tokens=True)[0]
+
+        try:
+            decoded_list = processor.tokenizer.batch_decode(new_tokens, skip_special_tokens=True)
+            decoded = decoded_list[0].strip() if decoded_list else "[No output generated]"
+        except Exception as decode_error:
+            decoded = f"[Decoding error on page {page_num}: {str(decode_error)}]"
+
 
         except Exception as e:
             decoded = f"[Error processing page {page_num}: {str(e)}]"