dynamic gpu

app.py

@@ -39,8 +39,8 @@ def init_pipeline():
     )
     pipe.to(device, dtype=dtype)
 
-
-def process_image_and_text(image, text, gemini_prompt, guidance, i_guidance, t_guidance):
+@spaces.GPU
+def generate_image(image, text, gemini_prompt, guidance, i_guidance, t_guidance):
     w, h, min_size = image.size[0], image.size[1], min(image.size)
     image = image.crop(
         ((w - min_size) // 2, (h - min_size) // 2, (w + min_size) // 2, (h + min_size) // 2)
@@ -119,7 +119,7 @@ with demo:
     )
 
     iface = gr.Interface(
-        fn=process_image_and_text,
+        fn=generate_image,
         inputs=[
             gr.Image(type="pil"),
             gr.Textbox(lines=2, label="text", info="Could be something as simple as 'this character playing soccer'."),