Update controlnet.py

controlnet.py

@@ -1,10 +1,11 @@
-import requests
+import subprocess
 import os
 
 base_url = "https://huggingface.co/nolanaatama/models/resolve/main/"
 output_directory = "/content/imagination/extensions/microsoftexcel-controlnet/models/"
 
 files_to_download = [
+    "control_v11e_sd15_ip2p_fp16.safetensors",
     "control_v11e_sd15_shuffle_fp16.safetensors",
     "control_v11p_sd15_canny_fp16.safetensors",
     "control_v11p_sd15_depth_fp16.safetensors",
@@ -34,13 +35,8 @@ for file_name in files_to_download:
     file_url = base_url + file_name
     save_path = os.path.join(output_directory, file_name)
     
-    response = requests.get(file_url, stream=True)
-    if response.status_code == 200:
-        with open(save_path, 'wb') as file:
-            for chunk in response.iter_content(chunk_size=8192):
-                file.write(chunk)
-        print(f"Downloaded {file_name}")
-    else:
-        print(f"Failed to download {file_name}")
+    subprocess.run(["wget", file_url, "-O", save_path])
+    
+    print(f"Downloaded {file_name}")
 
 print("All files downloaded successfully!")