Update app.py

app.py

@@ -29,8 +29,10 @@ def chdr(apol,prompt,modil,stips,fnamo,gaul):
         type="LDM3D"
         los=""
         tre='./tmpo/'+fnamo+'.json'
-        tra='./tmpo/'+fnamo+'_0.png'
-        trm='./tmpo/'+fnamo+'_1.png'
+        tra='./tmpo/'+fnamo+'_rgb_0.png'
+        trm='./tmpo/'+fnamo+'_rgb_1.png'
+        trh='./tmpo/'+fnamo+'_dep_0.png'
+        trv='./tmpo/'+fnamo+'_dep_1.png'
         flng=["yssup", "sllab", "stsaerb", "sinep", "selppin", "ssa", "tnuc", "mub", "kcoc", "kcid", "anigav", "dekan", "edun", "slatineg", "xes", "nrop", "stit", "ttub", "bojwolb", "noitartenep", "kcuf", "kcus", "kcil", "elttil", "gnuoy", "thgit", "lrig", "etitep", "dlihc", "yxes"]
         flng=[itm[::-1] for itm in flng]
         ptn = r"\b" + r"\b|\b".join(flng) + r"\b"
@@ -53,6 +55,8 @@ def chdr(apol,prompt,modil,stips,fnamo,gaul):
             del tre
             del tra
             del trm
+            del trh
+            del trv
         except:
             print("cant")
     except:
@@ -78,10 +82,10 @@ def plex(prompt,neg_prompt,stips,nut,wit,het,gaul,progress=gr.Progress(track_tqd
     image = pipe(prompt=[prompt]*2, negative_prompt=[neg_prompt]*2, generator=generator, guidance_scale=7.0, num_inference_steps=stips,height=het,width=wit)
     for a, imze in enumerate(image["rgb"]):
         apol.append(imze)
-        imze.save('./tmpo/'+fnamo+'_'+str(a)+'.png', 'PNG')
+        imze.save('./tmpo/'+fnamo+'_rgb_'+str(a)+'.png', 'PNG')
     for b, imbe in enumerate(image["depth"]):
         apol.append(imbe)
-        imbe.save('./tmpo/'+fnamo+'_'+str(b)+'.png', 'PNG')
+        imbe.save('./tmpo/'+fnamo+'_dep_'+str(b)+'.png', 'PNG')
     chdr(apol,prompt,modil,stips,fnamo,gaul)
     return apol