Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import re
-import spaces
+import spaces as spaces1, spaces2
 import random
 import string
 import torch
@@ -10,9 +10,9 @@ import numpy as np
 from lxml.html import fromstring
 #from transformers import pipeline
 from torch import multiprocessing as mp
-#from torch.multiprocessing import Pool as Pool3
-from pathos.multiprocessing import ProcessPool as Pool2
-from pathos.threading import ThreadPool as Pool1
+from torch.multiprocessing import Pool as Pool
+#from pathos.multiprocessing import ProcessPool as Pool
+#from pathos.threading import ThreadPool as Pool
 #from diffusers.pipelines.flux import FluxPipeline
 #from diffusers.utils import export_to_gif
 #from huggingface_hub import hf_hub_download
@@ -83,7 +83,7 @@ def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU(duration=40)
+@spaces1.GPU(duration=40)
 def Piper1(_do):
     print("starting piper1")
     retu = pp1(
@@ -97,7 +97,7 @@ def Piper1(_do):
     print("returning piper1")
     return retu
 
-@spaces.GPU(duration=40)
+@spaces2.GPU(duration=40)
 def Piper2(img,posi,neg):
     retu = pp2(
         prompt=posi,
@@ -106,7 +106,6 @@ def Piper2(img,posi,neg):
     )
     return retu
 
-@spaces.GPU(duration=40)
 def tok(pipe,txt):
     toks = pipe.tokenizer(txt)['input_ids']
     print(toks)
@@ -191,14 +190,14 @@ def run(p1,p2,*result):
         rng = list(range(ln))
     
         arr1 = [p for _ in rng]
-        pool1 = Pool1(ln)
+        pool1 = Pool(ln)
         out1 = list(pool1.imap(infer1,arr1))
         pool1.close()
         pool1.join()
         pool1.clear()
     
         arr2 = [{"a":p1_en,"b":p2_en,"c":out1[_]} for _ in rng]
-        pool2 = Pool2(ln)
+        pool2 = Pool(ln)
         out2 = list(pool2.imap(infer2,arr2))
         pool2.close()
         pool2.join()