Update app.py

app.py

@@ -57,6 +57,8 @@ def save_data(t,m,l,n,w,p):
     
     title = t.replace('https://',"").replace('/',"_").replace('.',"_")
     im = mi.run_1(name=title,img_size=[256,256])
+
+    im1 = f"{os.getcwd()}/images/{name}_prob.png"
     im2=f"https://{user_name}-{repo_name}.static.hf.space/images/{title}_prob.png"
     #im3=f"{make_im.root}/images/{name}_prob.png"
     #uid=uuid.uuid4()
@@ -76,7 +78,7 @@ def save_data(t,m,l,n,w,p):
         repo_type="space",
     )
     api.upload_file(
-        path_or_fileobj=im2,
+        path_or_fileobj=im1,
         path_in_repo=f"/images/{out_name}.png",
         repo_id=f'{user_name}/{repo_name}',
         token=token,