Update app.py

app.py

@@ -1,20 +1,25 @@
-import gradio as gr
+from huggingface_hub import login
 from diffusers import DiffusionPipeline
 import torch
+import gradio as gr
+import os
+
+# Faça login com seu token HF
+login(os.environ["HF_TOKEN"])  # use um token com permissão READ
+
+# Carrega o modelo
+pipe = DiffusionPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-schnell",
+    use_auth_token=True,  # garante que use autenticação
+    torch_dtype=torch.float16
+)
 
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float16)
 pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-def generate_image(prompt):
+def gerar(prompt):
     image = pipe(prompt).images[0]
     return image
 
-demo = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(label="Prompt"),
-    outputs=gr.Image(type="pil"),
-    title="FLUX.1 - Schnell",
-    description="Gere imagens com o modelo FLUX.1 da Black Forest Labs"
-)
-
+demo = gr.Interface(fn=gerar, inputs=gr.Textbox(), outputs="image")
 demo.launch()
+