Update app.py

app.py

@@ -10,8 +10,8 @@ from translatepy import Translator
 translator = Translator()
 
 # Constants
-base = "Corcelio/openvision"
-vae = "madebyollin/sdxl-vae-fp16-fix"
+model = "Corcelio/openvision"
+vae_model = "madebyollin/sdxl-vae-fp16-fix"
 
 CSS = """
 .gradio-container {
@@ -31,13 +31,13 @@ JS = """function () {
 
 # Load VAE component
 vae = AutoencoderKL.from_pretrained(
-    "madebyollin/sdxl-vae-fp16-fix", 
+    vae_model, 
     torch_dtype=torch.float16
 )
 
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
-    pipe = StableDiffusionXLPipeline.from_pretrained(base, vae=vae, torch_dtype=torch.float16).to("cuda")
+    pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to("cuda")