Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from cpn import CpnInterface
 from prep import multi_norm
@@ -7,6 +8,7 @@ from celldetection import label_cmap
 default_model = 'ginoro_CpnResNeXt101UNet-fbe875f1a3e5ce2c'
 
 
+@spaces.GPU
 def predict(filename, model=None, device=None, reduce_labels=True):
     global default_model
     assert isinstance(filename, str)