Upload folder using huggingface_hub

app.py

@@ -2,76 +2,37 @@ import gradio as gr
 import torch
 import spaces
 from diffusers import Flux2Pipeline
-from huggingface_hub import get_token
-import requests
-import io
 
 # Model configuration
-repo_id = "diffusers/FLUX.2-dev-bnb-4bit"
+repo_id = "black-forest-labs/FLUX.2-dev"
 torch_dtype = torch.bfloat16
 
-# Global pipeline variable
-pipe = None
-
-def download_model(progress=gr.Progress()):
-    """Download and cache the model before inference"""
-    global pipe
-    if pipe is not None:
-        return "✓ Model already loaded and ready!"
-    
-    progress(0.1, desc="Initializing download...")
-    progress(0.3, desc="Downloading model weights...")
-    
-    # Download model to CPU first (no GPU needed for download)
-    pipe = Flux2Pipeline.from_pretrained(
-        repo_id, text_encoder=None, torch_dtype=torch_dtype
-    )
-    
-    progress(1.0, desc="Download complete!")
-    return "✓ Model downloaded successfully! Ready to generate images."
-
-def remote_text_encoder(prompts):
-    response = requests.post(
-        "https://remote-text-encoder-flux-2.huggingface.co/predict",
-        json={"prompt": prompts},
-        headers={
-            "Authorization": f"Bearer {get_token()}",
-            "Content-Type": "application/json"
-        }
-    )
-    prompt_embeds = torch.load(io.BytesIO(response.content))
-    return prompt_embeds.to("cuda")
+# Download and initialize model at startup
+pipe = Flux2Pipeline.from_pretrained(
+    repo_id, torch_dtype=torch_dtype
+)
+pipe.enable_model_cpu_offload()
 
 @spaces.GPU(duration=120)
-def generate_image(prompt, progress=gr.Progress()):
-    global pipe
-    
+def generate_image(prompt, seed, steps, guidance_scale, progress=gr.Progress()):
     if not prompt:
         gr.Warning("Please enter a prompt")
         return None
     
-    if pipe is None:
-        gr.Warning("Please download the model first by clicking 'Download Model'")
-        return None
-    
-    progress(0.1, desc="Moving model to GPU...")
-    pipe = pipe.to("cuda")
-    
-    progress(0.3, desc="Encoding prompt...")
-    prompt_embeds = remote_text_encoder(prompt)
+    progress(0.2, desc="Preparing generation...")
     
-    generator = torch.Generator(device="cuda").manual_seed(42)
+    generator = torch.Generator(device="cpu").manual_seed(int(seed))
     
-    progress(0.5, desc="Generating...")
+    progress(0.4, desc="Generating image...")
     
     image = pipe(
-        prompt_embeds=prompt_embeds,
+        prompt=prompt,
         generator=generator,
-        num_inference_steps=50,
-        guidance_scale=4.0,
+        num_inference_steps=int(steps),
+        guidance_scale=guidance_scale,
     ).images[0]
     
-    progress(1.0, desc="Done")
+    progress(1.0, desc="Done!")
     return image
 
 # Minimal Apple-inspired CSS
@@ -150,33 +111,6 @@ apple_css = """
     background: #0077ed !important;
 }
 
-.download-btn {
-    background: #34c759 !important;
-    border: none !important;
-    border-radius: 980px !important;
-    padding: 16px 32px !important;
-    font-size: 17px !important;
-    font-weight: 400 !important;
-    color: white !important;
-    cursor: pointer !important;
-    transition: all 0.2s ease !important;
-    margin-top: 16px !important;
-}
-
-.download-btn:hover {
-    background: #30d158 !important;
-}
-
-.status-box {
-    background: #f5f5f7 !important;
-    border: none !important;
-    border-radius: 12px !important;
-    padding: 16px !important;
-    font-size: 15px !important;
-    color: #1d1d1f !important;
-    text-align: center !important;
-}
-
 .output-section {
     margin-top: 40px;
 }
@@ -204,7 +138,7 @@ apple_css = """
 }
 
 /* Hide labels for cleaner look */
-.input-section label, .output-section label {
+.input-section label {
     display: none !important;
 }
 
@@ -219,42 +153,59 @@ apple_css = """
     padding: 0 !important;
 }
 
-.button-row {
-    display: flex;
-    gap: 12px;
+.settings-section {
+    background: #f5f5f7 !important;
+    border-radius: 16px !important;
+    padding: 20px !important;
+    margin-top: 16px !important;
+}
+
+.settings-section label {
+    color: #1d1d1f !important;
+    font-size: 14px !important;
+    font-weight: 500 !important;
 }
 """
 
-with gr.Blocks(title="Image Studio") as demo:
+with gr.Blocks(title="FLUX.2 Image Studio") as demo:
     
     gr.HTML("""
         <div class="header-text">
-            <h1>Image Studio</h1>
-            <p>Describe what you imagine.</p>
+            <h1>FLUX.2 Studio</h1>
+            <p>Create stunning images with FLUX.2-dev</p>
         </div>
     """)
     
     with gr.Column(elem_classes="input-section"):
-        status_text = gr.Textbox(
-            value="⏳ Model not loaded. Click 'Download Model' to start.",
-            show_label=False,
-            interactive=False,
-            elem_classes="status-box"
-        )
-        
-        download_btn = gr.Button(
-            "Download Model",
-            elem_classes="download-btn",
-            variant="secondary"
-        )
-        
         prompt = gr.Textbox(
             placeholder="A peaceful mountain lake at sunrise...",
-            lines=2,
+            lines=3,
             show_label=False,
             container=False
         )
         
+        with gr.Accordion("Advanced Settings", open=False):
+            with gr.Row():
+                seed = gr.Number(
+                    label="Seed",
+                    value=42,
+                    precision=0
+                )
+                steps = gr.Slider(
+                    label="Inference Steps",
+                    minimum=10,
+                    maximum=100,
+                    value=50,
+                    step=1
+                )
+            guidance_scale = gr.Slider(
+                label="Guidance Scale",
+                minimum=1.0,
+                maximum=10.0,
+                value=4.0,
+                step=0.5
+            )
+        
         generate_btn = gr.Button(
             "Create",
             elem_classes="generate-btn",
@@ -274,23 +225,16 @@ with gr.Blocks(title="Image Studio") as demo:
         </div>
     """)
     
-    download_btn.click(
-        fn=download_model,
-        inputs=[],
-        outputs=[status_text],
-        api_visibility="private"
-    )
-    
     generate_btn.click(
         fn=generate_image,
-        inputs=[prompt],
+        inputs=[prompt, seed, steps, guidance_scale],
         outputs=[output_image],
         api_visibility="public"
     )
     
     prompt.submit(
         fn=generate_image,
-        inputs=[prompt],
+        inputs=[prompt, seed, steps, guidance_scale],
         outputs=[output_image],
         api_visibility="private"
     )

requirements.txt

@@ -1,15 +1,24 @@
 spaces
-huggingface_hub
 gradio
 torch
-requests
 git+https://github.com/huggingface/diffusers
 git+https://github.com/huggingface/transformers
 sentencepiece
 accelerate
 tokenizers
+datasets
+requests
+Pillow
 torchvision
 torchaudio
-Pillow
-datasets
-bitsandbytes
+numpy
+scipy
+opencv-python
+matplotlib
+safetensors
+omegaconf
+ftfy
+regex
+timm
+xformers
+invisible-watermark