Update app.py

app.py

@@ -36,6 +36,7 @@ import numpy as np
 from PIL import Image
 import torch
 
+import spaces
 import gradio as gr
 from huggingface_hub import snapshot_download
 
@@ -115,7 +116,7 @@ model = StreamMultiDiffusion(
     bootstrap_mix_steps=opt.bootstrap_steps,
     guidance_scale=opt.guidance_scale,
     seed=opt.seed,
-)
+).cuda()
 
 
 prompt_suggestions = [
@@ -380,6 +381,7 @@ def register(state, drawpad):
     return state
 
 
+@spaces.GPU
 def run(state, drawpad):
     state = register(state, drawpad)
     state.is_running = True
@@ -662,10 +664,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, head=head) as demo:
                 <img src='https://img.shields.io/badge/%F0%9F%A4%97%20Paper-StreamMultiDiffusion-yellow'>
             </a>
             &nbsp;
-            <a href='https://huggingface.co/spaces/ironjr/StreamMultiDiffusion'>
-                <img src='https://img.shields.io/badge/%F0%9F%A4%97%20Demo-StreamMultiDiffusion-yellow'>
-            </a>
-            &nbsp;
             <a href='https://huggingface.co/spaces/ironjr/SemanticPalette'>
                 <img src='https://img.shields.io/badge/%F0%9F%A4%97%20Demo-SemanticPaletteSD1.5-yellow'>
             </a>
@@ -673,10 +671,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, head=head) as demo:
             <a href='https://huggingface.co/spaces/ironjr/SemanticPaletteXL'>
                 <img src='https://img.shields.io/badge/%F0%9F%A4%97%20Demo-SemanticPaletteSDXL-yellow'>
             </a>
-            &nbsp;
-            <a href='https://colab.research.google.com/github/camenduru/SemanticPalette-jupyter/blob/main/SemanticPalette_jupyter.ipynb'>
-                <img src='https://colab.research.google.com/assets/colab-badge.svg'>
-            </a>
         </div>
     </div>
 </div>