Update app.py

app.py

@@ -42,21 +42,12 @@ def get_upscale_finegrain(prompt, img_path, upscale_factor):
         return None
 
 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)
-    if image is None:
-        return [None, None]
-    
-    image_path = "temp_image.jpg"
-    image.save(image_path, format="JPEG")
-    
+    ...
     if process_upscale:
-        upscale_image = get_upscale_finegrain(prompt, image_path, upscale_factor)
-        upscale_image_path = "upscale_image.jpg"
-        upscale_image.save(upscale_image_path, format="JPEG")
-        return [image_path, upscale_image_path]
-    else:
-        return [image_path, image_path]
+        upscale_image_path = get_upscale_finegrain(prompt, image_path, upscale_factor)
+        upscale_image = Image.open(upscale_image_path)  # Abrir la imagen utilizando PIL
+        upscale_image.save("upscale_image.jpg", format="JPEG")  # Guardar la imagen
+        return [image_path, "upscale_image.jpg"]
 
 css = """
 #col-container{ margin: 0 auto; max-width: 1024px;}