Update app.py

app.py

@@ -21,6 +21,7 @@ for i,model in enumerate(models):
 print (loaded_model)
 
 pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", safety_checker=None, variant="fp16", use_safetensors=True).to("cpu")
+pipeline.unet = torch.compile(pipeline.unet)
 
 def load_model(model_drop):
     pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32, use_safetensors=True)