Update app.py

app.py

@@ -22,8 +22,6 @@ from diffusers.utils import load_image
 #import jax
 #import jax.numpy as jnp
 
-result=[]
-
 def port_inc():
     env = os.getenv("CUSTOM_PORT")
     if env == None:
@@ -44,9 +42,6 @@ def pipe_i2i():
     PIPE.unet = torch.compile(PIPE.unet, mode="reduce-overhead", fullgraph=True)
     return PIPE
 
-pp1=pipe_t2i()
-pp2=pipe_i2i()
-
 def translate(text,lang):
     if text == None or lang == None:
         return ""       
@@ -94,19 +89,17 @@ def generate_random_string(length):
 
 @spaces.GPU(duration=40)
 def Piper1(_do):
-    try:
-        retu = pp1(
-            _do,
-            height=512,
-            width=512,
-            num_inference_steps=4,
-            max_sequence_length=256,
-            guidance_scale=0
-        )
-        return retu
-    except Exception as e:
-        print(e)
-        return None
+    print("starting piper1")
+    retu = pp1(
+        _do,
+        height=512,
+        width=512,
+        num_inference_steps=4,
+        max_sequence_length=256,
+        guidance_scale=0
+    )
+    print("returning piper1")
+    return retu
 
 @spaces.GPU(duration=40)
 def Piper2(img,posi,neg):
@@ -245,6 +238,14 @@ def ui():
 
 if __name__ == "__main__":
 
+    global result
+    global pp1
+    glibal pp2
+
+    result=[]
+    pp1=pipe_t2i()
+    pp2=pipe_i2i()
+
     mp.set_start_method("fork", force=True)
     port_inc()
     ui()