Update app.py

app.py

@@ -2,7 +2,7 @@ from diffusers import DiffusionPipeline
 import torch
 import gradio as gr
 from PIL import Image
-import os, random, gc, re, json, time
+import os, random, gc, re, json, time, shutil
 import PIL.Image
 import tqdm
 from accelerate import Accelerator
@@ -65,6 +65,7 @@ def chdr(apol,prompt,modil,los,stips,gaul):
         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)
+        shutil.rmtree('./tmpo')
     except:
         print("failed to umake obj")