Update app.py

app.py

@@ -238,7 +238,8 @@ if __name__ == "__main__":
 
     # Pre-load all models
     original_models, fine_models = load_model(options)
-    
+
+    @spaces.GPU
     def fit3d(image, model_option, kmeans_num):
         
         # Select model