Update app.py

app.py

@@ -34,15 +34,17 @@ async def generate_image(prompt, model, lora_word, width, height, scales, steps,
 async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model, process_lora):
     model = enable_lora(lora_model, basemodel) if process_lora else basemodel
     image, seed = await generate_image(prompt, model, "", width, height, scales, steps, seed)
-    image_path = "temp_image.png"
-    image.save(image_path)
+    image_path = "temp_image.jpg"
+    image.save(image_path, quality=95)
     
     if process_upscale:
+        upscale_image_path = "upscale_image.jpg"
         upscale_image = get_upscale_finegrain(prompt, image_path, upscale_factor)
+        upscale_image.save(upscale_image_path, quality=95)
     else:
-        upscale_image = image_path
+        upscale_image_path = image_path
     
-    return [image_path, upscale_image]
+    return [image_path, upscale_image_path]
 
 def get_upscale_finegrain(prompt, img_path, upscale_factor):
     client = Client("finegrain/finegrain-image-enhancer", hf_token=HF_TOKEN_UPSCALER)
@@ -76,7 +78,7 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
                     scales = gr.Slider(label="Guidance", minimum=3.5, maximum=7, step=0.1, value=3.5)
                     steps = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=24)
                     seed = gr.Slider(label="Seeds", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
-                    
+
                 submit_btn = gr.Button("Submit", scale=1)
                 submit_btn.click(
                     fn=lambda: None,
@@ -87,5 +89,4 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
                     fn=gen,
                     inputs=[prompt, basemodel_choice, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model_choice, process_lora],
                     outputs=[output_res]
-                )
-demo.launch()
+                )