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