Update dalle.py

Agora-Lab-AI · Oct 21, 2023 · 5440c7f · 5440c7f
1 parent 3eca1ec
commit 5440c7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dalle3/dalle.py b/dalle3/dalle.py
@@ -121,7 +121,7 @@ def get_urls(self):
                 set(
                     [
                         element.get_attribute("src")
-                        for element in WebDriverWait(self.driver, 600).until(
+                        for element in WebDriverWait(self.driver, 1600).until(
                             EC.presence_of_all_elements_located((By.CLASS_NAME, "mimg"))
                         )
                     ]