Update app.py

app.py

@@ -12,7 +12,7 @@ import random
 translator = Translator()
 
 # Constants
-model = "aixonlab/flux.1-lumiere-alpha"
+model = "Freepik/flux.1-lite-8B-alpha"
 
 CSS = """
 .gradio-container {
@@ -27,8 +27,17 @@ MAX_SEED = np.iinfo(np.int32).max
 
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
-    pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
+    transformer = FluxTransformer2DModel.from_single_file(
+        "https://huggingface.co/aixonlab/flux.1-lumiere-alpha/blob/main/lumiere_flux_alpha-fp8.safetensors",
+        torch_dtype=torch.bfloat16
+    )
+    pipe = FluxPipeline.from_pretrained(
+        model, 
+        transformer=transformer,
+        torch_dtype=torch.bfloat16)
     pipe.to("cuda")
+    
+
 
 # Function 
 @spaces.GPU()