Update app.py

app.py

@@ -9,6 +9,7 @@ import gradio as gr
 import matplotlib.pyplot as plt
 import tempfile
 import os 
+import spaces 
 
 from huggingface_hub import snapshot_download
 
@@ -225,6 +226,7 @@ def predict_wsi(image):
     pred_labels /= counter_tensor
     return pred_labels
 
+@spaces.GPU(duration=120)
 def segment_image(image):
     # img = Image.open(image_path)
     img = image