Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 import torch.nn.functional as F
@@ -47,6 +48,7 @@ for example_name in examples_names:
     examples.append(example)
 np.random.shuffle(examples) # shuffle
 
+@spaces.GPU
 def predict(input_image:Image.Image, true_label:str):
     """Predict the label of the input_image"""
     face = mtcnn(input_image)