Update app.py

app.py

@@ -1,13 +1,15 @@
 import gradio as gr
 import numpy as np
 import random
+import spaces
+rom diffusers import DiffusionPipeline
 
 # import spaces #[uncomment to use ZeroGPU]
 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-dev"  # Replace to the model you would like to use
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16