Update app.py

app.py

@@ -29,11 +29,7 @@ def port_inc():
     else:
         os.environ["CUSTOM_PORT"]=str(int(env)+1)
 
-def init_pool(_1,_2):
-    global infer1
-    global infer2
-    infer1 = _1
-    infer2 = _2
+def init_pool():
     port_inc()
 
 def pipe_t2i():
@@ -90,7 +86,7 @@ def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU(duration=55)
+@spaces.GPU(duration=40)
 def Piper1(pipe,_do):
     try:
         retu = pipe(
@@ -106,7 +102,7 @@ def Piper1(pipe,_do):
         print(e)
         return None
 
-@spaces.GPU(duration=55)
+@spaces.GPU(duration=40)
 def Piper2(pipe,img,posi,neg):
     try:
         retu = pipe(
@@ -119,39 +115,12 @@ def Piper2(pipe,img,posi,neg):
         print(e)
         return None
 
-@spaces.GPU(duration=55)
+@spaces.GPU(duration=40)
 def tok(pipe,txt):
     toks = pipe.tokenizer(txt)['input_ids']
     print(toks)
     return toks
 
-def infer(p):
-    p1 = p["a"]
-    p2 = p["b"]
-    name = generate_random_string(12)+".png"
-    _do = ['beautiful', 'playful', 'photographed', 'realistic', 'dynamic poze', 'deep field', 'reasonable coloring', 'rough texture', 'best quality', 'focused']
-    if p1 != "":
-        _do.append(f'{p1}')
-    if p2 != "":
-        _dont = f'{p2} where in {p1}'
-        neg = _dont
-    else:
-        neg = None
-    output = Piper1(infer1,'A '+" ".join(_do))
-    if output == None:
-        return None
-    else:
-        output.images[0].save(name)
-    if neg == None:
-        return name
-    img = load_image(name).convert("RGB")
-    output2 = Piper2(infer2,img,p1,neg)
-    if output2 == None:
-        return None
-    else:
-        output2.images[0].save("_"+name)
-    return "_"+name
-
 css="""
 input, input::placeholder {
     text-align: center !important;
@@ -197,6 +166,33 @@ def main():
     pp2=pipe_i2i()
     
     result = []
+
+    def infer(p):
+        p1 = p["a"]
+        p2 = p["b"]
+        name = generate_random_string(12)+".png"
+        _do = ['beautiful', 'playful', 'photographed', 'realistic', 'dynamic poze', 'deep field', 'reasonable coloring', 'rough texture', 'best quality', 'focused']
+        if p1 != "":
+            _do.append(f'{p1}')
+        if p2 != "":
+            _dont = f'{p2} where in {p1}'
+            neg = _dont
+        else:
+            neg = None
+        output = Piper1(pp1,'A '+" ".join(_do))
+        if output == None:
+            return None
+        else:
+            output.images[0].save(name)
+        if neg == None:
+            return name
+        img = load_image(name).convert("RGB")
+        output2 = Piper2(pp2,img,p1,neg)
+        if output2 == None:
+            return None
+        else:
+            output2.images[0].save("_"+name)
+        return "_"+name
     
     def run(p1,p2,*result):
         p1_en = translate(p1,"english")
@@ -206,7 +202,7 @@ def main():
         rng = list(range(ln))
         arr = [p for _ in rng]
         out = None
-        with Pool(ln, initializer=init_pool, initargs=(pp1,pp2)) as pool:
+        with Pool(ln, initializer=init_pool) as pool:
             out = pool.imap(infer,arr)
         return list(out)
     port_inc()