Update app.py

app.py

@@ -1,3 +1,6 @@
+
+# built-in
+
 import os
 import subprocess
 import logging
@@ -7,7 +10,9 @@ import string
 import requests
 import sys
 import warnings
-    
+
+# external
+
 import spaces
 import torch
 import gradio as gr
@@ -28,6 +33,8 @@ from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, Eul
 #import jax.numpy as jnp
 from numba import jit,njit
 
+# logging
+
 warnings.filterwarnings("ignore")
 root = logging.getLogger()
 root.setLevel(logging.DEBUG)
@@ -42,6 +49,12 @@ formatter = logging.Formatter('\n >>> [%(levelname)s] %(asctime)s %(name)s: %(me
 handler2.setFormatter(formatter)
 root.addHandler(handler2)
 
+# spawn
+
+mp.set_start_method("spawn", force=True)
+
+# data
+
 last_motion=None
 dtype = torch.float16
 result=[]
@@ -95,7 +108,6 @@ footer {
     display: flex;
 }
 """
-
 js="""
 function custom(){
     document.querySelector("div#prompt input").setAttribute("maxlength","38")
@@ -103,6 +115,8 @@ function custom(){
 }
 """
 
+# functionality
+
 @njit(cache=True,nopython=True,parallel=True)
 def run(cmd, assert_success=False, capture_output=False, env=None, dry_run=False):
     if dry_run:
@@ -117,7 +131,7 @@ def run(cmd, assert_success=False, capture_output=False, env=None, dry_run=False
             sys.exit()
 
     return result
-
+    
 @njit(cache=True,nopython=True,parallel=True)
 def translate(text,lang):
     if text == None or lang == None:
@@ -159,12 +173,12 @@ def translate(text,lang):
     ret = re.sub(f'[{string.punctuation}]', '', re.sub('[\s+]', ' ', translated)).lower().strip()
     print(ret)
     return ret
-
+    
 @njit(cache=True,nopython=True,parallel=True)
 def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
-
+    
 #@spaces.GPU(duration=65)
 @njit(cache=True,nopython=True,parallel=True)
 def Piper(image,positive,negative,motion):
@@ -312,7 +326,10 @@ def pre():
     pipe.enable_vae_slicing()
     pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
 
-    mp.set_start_method("spawn", force=True)
+# entry
+
+if __name__ == "__main__":
+    pre()
+    ui()
 
-pre()
-ui()
+# end