Merge pull request #16 from d3n7/patch-3

Fix censorship detection
Agora-Lab-AI · Oct 21, 2023 · 290c6a0 · 290c6a0
2 parents 61dc6a3 + 8e447fc
commit 290c6a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dalle3/dalle.py b/dalle3/dalle.py
@@ -119,7 +119,7 @@ def get_urls(self):
         try:
             #check for 5 seconds if censored message comes up. if it doesn't, it goes to the next except block
             censored = WebDriverWait(self.driver, 5).until(
-                EC.presence_of_all_elements_located((By.CLASS_NAME, "gil_err_cp"))
+                EC.presence_of_all_elements_located((By.CLASS_NAME, "gil_err_mt"))
             )
             logging.critical("Prompt censored by OpenAI")
             return []