Update app.py (#3)

- Update app.py (aaf793d29adebed6d08998e3e918227a76a31c44)


Co-authored-by: Kun Yan <[email protected]>

app.py

@@ -4,6 +4,7 @@ import random
 
 from PIL import Image
 
+import spaces
 from http import HTTPStatus
 from urllib.parse import urlparse, unquote
 from pathlib import PurePosixPath
@@ -30,7 +31,7 @@ def get_image_size(aspect_ratio):
     else:
         return 1328, 1328
 
-
+@spaces.GPU(duration=65)
 def infer(
     prompt,
     negative_prompt=" ",
@@ -44,7 +45,7 @@ def infer(
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     width, height = get_image_size(aspect_ratio)
-
+    print("calling with prompt: %s" % prompt)
     rsp = ImageSynthesis.call(api_key=os.environ.get("DASH_API_KEY"),
                             model="qwen-image",
                             prompt=prompt,
@@ -155,7 +156,7 @@ with gr.Blocks(css=css) as demo:
                     value=50, 
                 )
 
-        gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=True, cache_mode="lazy")
+        gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=False, cache_mode="lazy")
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,