change to flux

app.py

@@ -7,7 +7,7 @@ from diffusers import DiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
+model_repo_id = "black-forest-labs/FLUX.1-schnell"  # Replace to the model you would like to use
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16