salomonsky commited on
Commit
812aaeb
1 Parent(s): 54bc2ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -76,14 +76,14 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
76
  upscale_image = Image.open(upscale_image_path)
77
  upscale_image.save(DATA_PATH / f"upscale_image_{seed}.jpg", format="JPEG")
78
  progress_bar.progress(100)
79
- image_path.unlink() # Remove the original image
80
- return [str(DATA_PATH / f"upscale_image_{seed}.jpg"), str(prompt_file_path")]
81
  else:
82
  progress_bar.empty()
83
- return [str(image_path), str(prompt_file_path")]
84
  else:
85
  progress_bar.progress(100)
86
- return [str(image_path), str(prompt_file_path")]
87
 
88
  async def improve_prompt(prompt):
89
  try:
 
76
  upscale_image = Image.open(upscale_image_path)
77
  upscale_image.save(DATA_PATH / f"upscale_image_{seed}.jpg", format="JPEG")
78
  progress_bar.progress(100)
79
+ image_path.unlink()
80
+ return [str(DATA_PATH / f"upscale_image_{seed}.jpg"), str(prompt_file_path)]
81
  else:
82
  progress_bar.empty()
83
+ return [str(image_path), str(prompt_file_path)]
84
  else:
85
  progress_bar.progress(100)
86
+ return [str(image_path), str(prompt_file_path)]
87
 
88
  async def improve_prompt(prompt):
89
  try: