salomonsky commited on
Commit
8221a06
1 Parent(s): b20c582

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -103,9 +103,11 @@ async def gen(
103
  progress=gr.Progress(track_tqdm=True)
104
  ):
105
  model = enable_lora(lora_add)
106
- print(model)
107
  image, seed = await generate_image(prompt,model,lora_word,width,height,scales,steps,seed)
108
- upscaled_image = await upscale_image(image, upscale_factor)
 
 
 
109
  return upscaled_image, seed
110
 
111
  with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
 
103
  progress=gr.Progress(track_tqdm=True)
104
  ):
105
  model = enable_lora(lora_add)
 
106
  image, seed = await generate_image(prompt,model,lora_word,width,height,scales,steps,seed)
107
+ image_path = "image.png"
108
+ image.save(image_path)
109
+
110
+ upscaled_image = await upscale_image(image_path, upscale_factor)
111
  return upscaled_image, seed
112
 
113
  with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo: