Update app.py

app.py

@@ -4,7 +4,6 @@ from PIL import Image
 import numpy as np
 import time
 import os
-from huggingface_hub import spaces as hf_spaces
 
 # --- Import Custom Pipelines ---
 # Ensure these files are in the same directory or accessible in PYTHONPATH
@@ -78,7 +77,7 @@ def get_mae_pipeline():
     return mae_pipe_global
 
 # --- Unified Prediction Function ---
-@hf_spaces.GPU(duration=90)
+@spaces.GPU(duration=90)
 def segment_image(input_image: Image.Image, model_choice: str) -> Image.Image:
     """
     Takes a PIL Image and model choice, performs segmentation, and returns the segmented image.