Update app.py

app.py

@@ -58,7 +58,7 @@ 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"
+    im1 = f"{os.getcwd()}/images/{title}_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()