Update app.py

app.py

@@ -22,14 +22,15 @@ def demo_process(input_pdf):
         f.write(image_bytes)
 
     # zip_path = "images.zip"
-    # image_path = os.path.join(temp_dir, image_name)
+    image_path = os.path.join(temp_dir, image_name)
     # output = model.inference(image=image_name, prompt=task_prompt)["predictions"][0]    
     # with zipfile.ZipFile(zip_path, "w") as zipf:
     #     zipf.write(os.path.join(temp_dir, image_name), image_name)    
     # os.remove(image_path)
     # os.rmdir(temp_dir)
+    image_path = Image.fromarray(image_path)
     
-    return image_name
+    return image_path
     
 
 task_name = "SGSInvoice"