Update app.py

app.py

@@ -58,11 +58,12 @@ var_2="nota-ai/bk-sdm-small"
 
 
 pipe = StableDiffusionPipeline.from_pretrained(
-    var_1, torch_dtype=torch_dtype, use_safetensors=True)
+     model_repo_id, torch_dtype=torch_dtype, use_safetensors=True)
 #pipe.vae = AutoencoderTiny.from_pretrained(
 #    "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
 pipe = pipe.to(device)
 pipe.enable_vae_tiling()
+pipe.enable_sequential_cpu_offload()
 
 @spaces.GPU(duration=25) #[uncomment to use ZeroGPU]
 def infer(