Update app.py

app.py

@@ -3,39 +3,6 @@ import subprocess
 import sys
 import io
 from kernels import get_kernel
-
-# Clone and install diffusers from private repo
-def setup_diffusers():
-    gh_token = os.environ.get("GH_API_KEY")
-    if not gh_token:
-        raise ValueError("GH_API_KEY environment variable not set")
-    
-    repo_url = f"https://{gh_token}@github.com/apolinario/diffusers-new-model-addition.git"
-    repo_dir = "diffusers"
-    
-    # Clone repo if it doesn't exist
-    if not os.path.exists(repo_dir):
-        print("Cloning diffusers repository...")
-        subprocess.run(["git", "clone", repo_url, repo_dir], check=True)
-    
-    # Checkout the branch
-    #print("Checking out full-flux2 branch...")
-    #subprocess.run(["git", "-C", repo_dir, "checkout", "full-flux2"], check=True)
-        
-    # Install in editable mode
-    print("Installing diffusers in editable mode...")
-    subprocess.run([sys.executable, "-m", "pip", "install", "-e", repo_dir], check=True)
-
-    # Add to Python path
-    diffusers_src = os.path.abspath(os.path.join(repo_dir, "src"))
-    if diffusers_src not in sys.path:
-        sys.path.insert(0, diffusers_src)
-    
-    print("Diffusers setup complete!")
-
-# Run setup
-setup_diffusers()
-
 import gradio as gr
 import numpy as np
 import random
@@ -73,7 +40,8 @@ def remote_text_encoder(prompts):
 # Load model
 fa3_kernel = get_kernel("kernels-community/flash-attn3", revision="fake-ops-return-probs")
 
-repo_id = "diffusers-internal-dev/new-model-image-final-weights"
+repo_id = "black-forest-labs/FLUX.2-dev"
+
 dit = Flux2Transformer2DModel.from_pretrained(
     repo_id,
     subfolder="transformer",