Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 import numpy as np
+from rembg import remove
 import random
 from huggingface_hub import AsyncInferenceClient
 from translatepy import Translator
@@ -63,7 +64,7 @@ async def generate_image(prompt, model, lora_word, width, height, scales, steps,
         print(f"[-] Generating image with prompt: {text}, model: {model}")
         client = AsyncInferenceClient()
         image = await client.text_to_image(prompt=text, height=height, width=width, guidance_scale=scales, num_inference_steps=steps, model=model)
-        return image, seed
+        return remove(image), seed
     except Exception as e:
         print(f"[-] Error generating image: {e}")
         return None, None