Update app.py

app.py

@@ -22,7 +22,7 @@ from diffusers.utils import load_image
 #import jax.numpy as jnp
 import torch._dynamo
 
-#set_start_method("spawn", force=True)
+set_start_method("spawn", force=True)
 torch._dynamo.config.suppress_errors = True
 
 #pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, revision="refs/pr/1", token=os.getenv("hf_token")).to(device)
@@ -89,7 +89,6 @@ def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU(duration=35)
 def Piper(_do):
     pipe = pipe_t2i()
     try:
@@ -106,7 +105,6 @@ def Piper(_do):
         print(e)
         return None
 
-@spaces.GPU(duration=35)
 def Piper2(img,posi,neg):
     pipe = pipe_i2i()
     try:
@@ -120,12 +118,12 @@ def Piper2(img,posi,neg):
         print(e)
         return None
 
-@spaces.GPU(duration=35)
 def tok(txt):
     toks = pipe.tokenizer(txt)['input_ids']
     print(toks)
     return toks
-    
+
+@spaces.GPU(duration=75)
 def infer(p1,p2):
     name = generate_random_string(12)+".png"
     _do = ['beautiful', 'playful', 'photographed', 'realistic', 'dynamic poze', 'deep field', 'reasonable coloring', 'rough texture', 'best quality', 'focused']