Update screenshot.py

screenshot.py

@@ -2,6 +2,7 @@ from selenium import webdriver
 from selenium.common.exceptions import WebDriverException
 from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.common.by import By
+import urllib
 #from selenium_webdriver import WebElement
 import uuid
 
@@ -14,7 +15,7 @@ def create_ss(graph,width=1000,height=8000):
     for ea in graph.split("\n"):
         code=ea.strip().strip("\n")
         code_out+=code
-    url=f"https://omnibus-text-to-chart.hf.space/file=merm.html?mermaid={code_out}"
+    url=f"https://omnibus-mermaid-script.static.hf.space/index.html?mermaid={urllib.parse.quote_plus(code_out)}"
     is_url=True
     wait_time=10000
     if is_url: