Update app.py

app.py

@@ -8,9 +8,7 @@ if not os.path.exists(os.getenv("checkpoint_path")):
     os.system("git clone --recurse-submodules https://github.com/ztxz16/fastllm.git")
     os.system("cd fastllm; mkdir build; cd build; cmake ..; make -j; cd tools; python setup.py install --user --prefix=")
     os.system("wget https://huggingface.co/huangyuyang/chatglm2-6b-int4.flm/resolve/main/chatglm2-6b-int4.flm")
-    my_env = os.environ.copy()
-    # my_env["PATH"] = f"/usr/sbin:/sbin:{my_env['PATH']}"
-    subprocess.Popen("uvicorn api:app --reload", env=my_env)
+    subprocess.Popen("uvicorn", "api:app")
 
 class GUI:
     def __init__(self, *args, **kwargs):
@@ -34,12 +32,13 @@ class GUI:
         chapter_titles = epub2txt.content_titles
         title = epub2txt.title
         if self.ws is None:
-            self.ws = create_connection(f"ws://localhost:8000/{file.name}/ws")
+            self.ws = create_connection(f"ws://localhost:8000/ws")
+            self.ws.send(file.name)
         res = ''
         while 'output: ' not in res:
             res = self.ws.recv()
             if 'chsum: ' in res:
-                self.out.append(res.remove("chsum: "))
+                self.out.append(res.replace("chsum: ", ""))
             elif 'draft_sum: ' in res:
                 h = res[11:]
             elif 'output: ' in res: