Update app.py

app.py

@@ -147,6 +147,23 @@ def generate_image(prompt, image, image2):
                 process.kill()
     return generate_image(prompt, image, image2)
 
+
+@spaces.GPU(duration=50)
+def generate_image_50(prompt, image, image2):
+  return generate_image(prompt, image, image2)
+
+@spaces.GPU(duration=70)
+def generate_image_70(prompt, image, image2):
+  return generate_image(prompt, image, image2)
+
+@spaces.GPU(duration=90)
+def generate_image_90(prompt, image, image2):
+  return generate_image(prompt, image, image2)
+
+@spaces.GPU(duration=110)
+def generate_image_110(prompt, image, image2):
+  return generate_image(prompt, image, image2)
+
 @spaces.GPU(duration=130)
 def generate_image_130(prompt, image, image2):
   return generate_image(prompt, image, image2)
@@ -167,8 +184,32 @@ def generate_image_190(prompt, image, image2):
 def generate_image_200(prompt, image, image2):
   return generate_image(prompt, image, image2)
 
+@spaces.GPU(duration=210)
+def generate_image_210(prompt, image, image2):
+  return generate_image(prompt, image, image2)
+
+@spaces.GPU(duration=220)
+def generate_image_220(prompt, image, image2):
+  return generate_image(prompt, image, image2)
+
+@spaces.GPU(duration=230)
+def generate_image_230(prompt, image, image2):
+  return generate_image(prompt, image, image2)
+
+@spaces.GPU(duration=240)
+def generate_image_240(prompt, image, image2):
+  return generate_image(prompt, image, image2)
+    
 def generate_image_wrapper(prompt, image, image2, duration):
-  if duration == "130":
+  if duration == "50":
+    return generate_image_50(prompt, image, image2)
+  elif duration == "70":
+    return generate_image_70(prompt, image, image2)
+  elif duration == "90":
+    return generate_image_90(prompt, image, image2)
+  elif duration == "110":
+    return generate_image_110(prompt, image, image2)
+  elif duration == "130":
     return generate_image_130(prompt, image, image2)
   elif duration == "150":
     return generate_image_150(prompt, image, image2)
@@ -178,6 +219,14 @@ def generate_image_wrapper(prompt, image, image2, duration):
     return generate_image_190(prompt, image, image2)
   elif duration == "200":
     return generate_image_200(prompt, image, image2)
+  elif duration == "210":
+    return generate_image_210(prompt, image, image2)
+  elif duration == "220":
+    return generate_image_220(prompt, image, image2)
+  elif duration == "230":
+    return generate_image_230(prompt, image, image2)
+  elif duration == "240":
+    return generate_image_240(prompt, image, image2)
   else:
     return generate_image_170(prompt, image, image2)
 
@@ -187,7 +236,7 @@ if __name__ == "__main__":
                             "text",
                             gr.Image(image_mode='RGBA', type="numpy"),
                             gr.Image(image_mode='RGBA', type="numpy"),
-                            gr.Radio(["130", "150", "170", "190", "200"], label="GPU acclocation in secs")
+                            "text"
                         ],
                         outputs=[
                             gr.Image(type="numpy", image_mode='RGBA')