Rooni commited on
Commit
fc4ed9a
·
verified ·
1 Parent(s): b75f00d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -51,7 +51,7 @@ def generate_image(prompt, structure_image, style_image, depth_strength=15, styl
51
  structure_image_url = upload_image_to_hf_cache(structure_image)
52
  style_image_url = upload_image_to_hf_cache(style_image)
53
 
54
- client = Client("multimodalart/flux-style-shaping", huggingface_token=get_random_api_key())
55
  result = client.predict(
56
  prompt=prompt,
57
  structure_image=handle_file(structure_image_url),
@@ -90,8 +90,7 @@ with gr.Blocks(css=css) as app:
90
  inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
91
  outputs=[output_image],
92
  fn=generate_image,
93
- cache_examples=True,
94
- cache_mode="lazy"
95
  )
96
 
97
  with gr.Column():
 
51
  structure_image_url = upload_image_to_hf_cache(structure_image)
52
  style_image_url = upload_image_to_hf_cache(style_image)
53
 
54
+ client = Client("multimodalart/flux-style-shaping", hf_token=get_random_api_key())
55
  result = client.predict(
56
  prompt=prompt,
57
  structure_image=handle_file(structure_image_url),
 
90
  inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
91
  outputs=[output_image],
92
  fn=generate_image,
93
+ cache_examples=False,
 
94
  )
95
 
96
  with gr.Column():