Update app.py

app.py

@@ -20,19 +20,16 @@ def demo_process(input_pdf):
     
     with open(os.path.join(temp_dir, image_name), "wb") as f:
         f.write(image_bytes)
-        
-
 
     # zip_path = "images.zip"
     # image_path = os.path.join(temp_dir, image_name)
-    output = model.inference(image=image_name, prompt=task_prompt)["predictions"][0]    
+    # 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)
+    #     zipf.write(os.path.join(temp_dir, image_name), image_name)    
+    # os.remove(image_path)
+    # os.rmdir(temp_dir)
     
-    return output
+    return image_name
     
 
 task_name = "SGSInvoice"