Update app.py

app.py

@@ -4,10 +4,7 @@ import subprocess
 import importlib
 
 def setup_environment():
-    # Update pip
     subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "pip"])
-    
-    # Install or upgrade gradio and pygit2
     subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "gradio==4.29.0", "pygit2==1.12.2"])
 
 def clone_repo():
@@ -16,8 +13,19 @@ def clone_repo():
         subprocess.check_call(["git", "clone", repo_url])
     os.chdir("Automated_DeFooocus")
     
-    # Install requirements, ignoring conflicts
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements_versions.txt", "--ignore-installed"])
+    # Install requirements, ignoring torchsde
+    with open("requirements_versions.txt", "r") as f:
+        requirements = f.read().splitlines()
+    
+    filtered_requirements = [req for req in requirements if not req.startswith("torchsde")]
+    
+    with open("temp_requirements.txt", "w") as f:
+        f.write("\n".join(filtered_requirements))
+    
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "temp_requirements.txt", "--ignore-installed"])
+    
+    # Install torchsde separately
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "torchsde==0.2.6", "--no-deps"])
 
 def run_fooocus(theme="dark", preset="default", advanced_args="--attention-split --always-high-vram --disable-offload-from-vram --all-in-fp16"):
     args = f"{advanced_args} --theme {theme}"
@@ -27,7 +35,6 @@ def run_fooocus(theme="dark", preset="default", advanced_args="--attention-split
     subprocess.Popen([sys.executable, "entry_with_update.py"] + args.split())
 
 def launch_interface():
-    # Import gradio here to ensure we're using the correct version
     import gradio as gr
     
     with gr.Blocks() as demo:
@@ -47,9 +54,5 @@ if __name__ == "__main__":
     setup_environment()
     clone_repo()
     print("[DeFooocus] Starting ...")
-    
-    # Reload the gradio module to ensure we're using the correct version
-    importlib.reload(sys.modules['gradio'])
-    
     demo = launch_interface()
     demo.launch()