Update app.py

app.py

@@ -199,7 +199,7 @@ def flux_pipe_call_that_returns_an_iterable_of_images(
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
-base_model = "/media/vishnuanilkumar/New Volume/FLUX.1-dev"
+base_model = "black-forest-labs/FLUX.1-dev"
 
 #TAEF1 is very tiny autoencoder which uses the same "latent API" as FLUX.1's VAE. FLUX.1 is useful for real-time previewing of the FLUX.1 generation process.#
 taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)