Update app.py

app.py

@@ -130,8 +130,8 @@ def generate_annotated_image(dataset_name, split, domain, category):
 
   if len(msg_error) > 0:
     # save 
-    images_wo_content.save("img_paragraphs.png")
-    images_wo_content.save("img_lines.png")
+    Image.open(images_wo_content).save("img_paragraphs.png")
+    Image.open(images_wo_content).save("img_lines.png")
     df_paragraphs_wo_content.to_csv("paragraphs.csv", encoding="utf-8", index=False)
     df_lines_wo_content.to_csv("lines.csv", encoding="utf-8", index=False)
     return msg_error, images_wo_content, images_wo_content, "img_paragraphs.png", "img_lines.png", df_paragraphs_wo_content, df_lines_wo_content, gr.File.update(value="paragraphs.csv", visible=False), gr.File.update(value="lines.csv", visible=False)