Update app.py

app.py

@@ -13,7 +13,6 @@ translator = Translator()
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 # Constants
 model = "stabilityai/stable-diffusion-3-medium"
-vae_model = "madebyollin/sdxl-vae-fp16-fix"
 
 
 model_path = snapshot_download(
@@ -44,15 +43,11 @@ JS = """function () {
 }"""
 
 
-# Load VAE component
-vae = AutoencoderKL.from_pretrained(
-    vae_model, 
-    torch_dtype=torch.float16
-)
+
 
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
-    pipe = StableDiffusion3Pipeline.from_pretrained(model_path, vae=vae, torch_dtype=torch.float16).to("cuda")
+    pipe = StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16).to("cuda")
 
 
 # Function 
@@ -97,7 +92,7 @@ examples = [
 
 with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     gr.HTML("<h1><center>SD3M🦄</center></h1>")
-    gr.HTML("<p><center><a href='https://huggingface.co/Corcelio/mobius'>mobius</a> text-to-image generation</center><br><center>Multi-Languages. Adding default prompts to enhance.</center></p>")
+    gr.HTML("<p><center><a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium'>sd3m</a> text-to-image generation</center><br><center>Multi-Languages. Adding default prompts to enhance.</center></p>")
     with gr.Group():
         with gr.Row():
             prompt = gr.Textbox(label='Enter Your Prompt', value="best quality, HD, aesthetic", scale=6)