Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 import numpy as np
@@ -58,27 +59,27 @@ css = """
 }
 """
 
-@gr.Space_GPU
+@spaces.GPU
 def run_extract_features():
     return extract_features()
 
-@gr.Space_GPU
+@spaces.GPU
 def run_find_optimal_clusters(ga, min_clusters, max_clusters):
     return find_optimal_clusters(ga, min_clusters, max_clusters)
 
-@gr.Space_GPU
+@spaces.GPU
 def run_perform_clustering(ga, num_clusters):
     return perform_clustering(ga, num_clusters)
 
-@gr.Space_GPU
+@spaces.GPU
 def run_plot_kaplan_meier(ga):
     return plot_kaplan_meier(ga)
 
-@gr.Space_GPU
+@spaces.GPU
 def run_plot_median_survival_bar(ga):
     return plot_median_survival_bar(ga)
 
-@gr.Space_GPU
+@spaces.GPU
 def run_perform_log_rank_test(ga):
     return perform_log_rank_test(ga)