Update app.py

app.py

@@ -3,9 +3,9 @@ from diffusers import FluxPipeline
 import gradio as gr
 
 def generate_flux_image(
-    prompt, 
-    width=512, 
-    height=512, 
+    prompt,
+    width=512,
+    height=512,
     num_inference_steps=50,
     guidance_scale=7.5,
     seed=None
@@ -18,7 +18,7 @@ def generate_flux_image(
     
     # Pipeline laden
     pipeline = FluxPipeline.from_pretrained(
-        "improvements/flux",  # Passen Sie den Pfad an
+        "models/enhanceaiteam/Flux-uncensored",  # Korrigierter Pfad
         torch_dtype=torch.float16
     )
     
@@ -55,7 +55,7 @@ def create_gradio_interface():
         output_image = gr.Image(label="Generiertes Bild")
         
         generate_btn.click(
-            fn=generate_flux_image, 
+            fn=generate_flux_image,
             inputs=[prompt, width, height, steps, guidance, seed],
             outputs=output_image
         )