Update app.py

app.py

@@ -129,7 +129,7 @@ def main():
                     st.error(f"An error occurred: {e}")
                     
             elif selected_model == "stabilityai/stable-diffusion-xl-base-1.0" or user_input == '/image:':
-                pipe = DiffusionPipeline.from_pretrained(models, torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
+                pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
                 pipe.to("cuda")
 
                 params = {'num_inference_steps': 100, 'num_images_per_prompt': 2}