Update app.py

app.py

@@ -34,17 +34,15 @@ 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.jpg"
-    image.save(image_path, quality=95)
+    image_path = "temp_image.jpg"  
+    image.save(image_path, format="JPEG")  
     
     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_path = image_path
-    
-    return [image_path, upscale_image_path]
+        upscale_image = image_path
+
+    return [image_path, upscale_image]
 
 def get_upscale_finegrain(prompt, img_path, upscale_factor):
     client = Client("finegrain/finegrain-image-enhancer", hf_token=HF_TOKEN_UPSCALER)