Update app.py

app.py

@@ -122,6 +122,11 @@ def chdr(apol,prompt,modil,los,stips,fnamo,gaul):
                 json.dump(dobj, f)
             HfApi.upload_folder(repo_id="JoPmt/hf_community_images",folder_path="./tmpo",repo_type="dataset",path_in_repo="./",token=HF_TOKEN)
         dobj={'img_name':fnamo,'model':modil,'lora':los,'prompt':prompt,'steps':stips,'type':type,'haed':gaul,}
+        try:
+            for pxn in glob.glob('./tmpo/*.png'):
+                os.remove(pxn)
+        except:
+            print("mar")
         with open(tre, 'w') as f:
             json.dump(dobj, f)
         HfApi.upload_folder(repo_id="JoPmt/Tst_datast_imgs",folder_path="./tmpo",repo_type="dataset",path_in_repo="./",token=HF_TOKEN)