Update app.py

app.py

@@ -33,7 +33,7 @@ def init_pool():
     port_inc()
 
 def pipe_t2i():
-    PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token"), device=-1)
+    PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token"))
     return PIPE
     
 def pipe_i2i():
@@ -161,6 +161,7 @@ function custom(){
 """
 
 def infer(p):
+        print("infer: started")
         p1 = p["a"]
         p2 = p["b"]
         name = generate_random_string(12)+".png"
@@ -192,9 +193,11 @@ def run(p1,p2,*result):
         p2_en = translate(p2,"english")
         p = {"a":p1_en,"b":p2_en}
         ln = len(result)
+        print("images: "+str(ln))
         rng = list(range(ln))
         arr = [p for _ in rng]
         out = None
+        ptint("Starting pool!")
         with Pool(ln, initializer=init_pool) as pool:
             out = pool.imap(infer,arr)
         return list(out)