Update modelos.py

modelos.py

@@ -103,10 +103,21 @@ def d_b_click(b):
     
     with garis3:
         display(HTML(gariz3))
-        
+
+    # Define the two directory paths
+    dir_path_1 = "/home/studio-lab-user/forge/models/Stable-diffusion/tmp_models"
+    dir_path_2 = "/home/studio-lab-user/asd/models/Stable-diffusion/tmp_models"
+
+    # Check if the directory exists at the first location
+    if os.path.exists(dir_path_1):
+        directory_path = dir_path_1
+    else:
+        directory_path = dir_path_2
+
+    # Change the current directory to the selected directory
     with dbo:
         say("【{red} Downloading{cyan} Models{magenta} Models{yellow} 】{red}")
-        os.chdir("/home/studio-lab-user/forge/models/Stable-diffusion/tmp_models")
+        os.chdir(directory_path)
         
         for url in surl:
             download(url)