IllusionDiffusion

Running on Zero

App Files Files Community

AP123

cbensimon HF staff commited on Jan 22

Commit

18274c1

•

1 Parent(s): 8e6c97d

ZeroGPU (#796)

Browse files

- ZeroGPU (4bdf74fe79687ebb8544720dd838eee0d3a153c0)
- Update app.py (12d957514d87b6a185b1c858c975a0a07ca89a5f)

Co-authored-by: Charles Bensimon <cbensimon@users.noreply.huggingface.co>

Files changed (2) hide show

README.md +1 -1
app.py +3 -1

README.md CHANGED Viewed

@@ -4,7 +4,7 @@ emoji: 👁
 colorFrom: red
 colorTo: pink
 sdk: gradio
-sdk_version: 3.44.3
 app_file: app.py
 pinned: false
 license: openrail

 colorFrom: red
 colorTo: pink
 sdk: gradio
+sdk_version: 4.14.0
 app_file: app.py
 pinned: false
 license: openrail

app.py CHANGED Viewed

@@ -1,3 +1,4 @@
 import torch
 import gradio as gr
 from gradio import processing_utils, utils
@@ -106,6 +107,7 @@ def convert_to_base64(pil_image):
     return base64_image
 # Inference function
 def inference(
     control_image: Image.Image,
     prompt: str,
@@ -267,7 +269,7 @@ with gr.Blocks() as app:
         queue=False,
         postprocess=False
     )
-    share_button.click(None, [], [], _js=share_js)
 with gr.Blocks(css=css) as app_with_history:
     with gr.Tab("Demo"):

+import spaces
 import torch
 import gradio as gr
 from gradio import processing_utils, utils
     return base64_image
 # Inference function
+@spaces.GPU
 def inference(
     control_image: Image.Image,
     prompt: str,
         queue=False,
         postprocess=False
     )
+    share_button.click(None, [], [], js=share_js)
 with gr.Blocks(css=css) as app_with_history:
     with gr.Tab("Demo"):