Update app.py

app.py

@@ -57,12 +57,8 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             page = driver.find_element(By.TAG_NAME, "html")
             print (dir(page))
             print (page.text)
-
-            js="""
-            localStorage.setItem('theme', 'gab-black');
-            """
-            
-            driver.execute_script(js)
+           
+            driver.execute_script("localStorage.setItem('theme', 'gab-black');")
             main_head = driver.find_element(By.CLASS_NAME, "main-content-header")
             head_shot = main_head.screenshot(f'head-{uid}-tmp.png')