update app.py

app.py

@@ -8,10 +8,10 @@ import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
-model_repo = "black-forest-labs/FLUX.1-dev"  # Replace to the model you would like to use
+model_repo = "models/black-forest-labs/FLUX.1-dev"  # Replace to the model you would like to use
 
 pipe = FluxPipeline.from_pretrained(model_repo, torch_dtype=torch_dtype).to(device)
-pipe.enable_model_cpu_offload()
+#pipe.enable_model_cpu_offload()
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024