Update app.py

app.py

@@ -7,6 +7,7 @@ from sklearn.preprocessing import MinMaxScaler
 import tensorflow as tf
 from tensorflow.keras import layers, models, regularizers
 import os
+import spaces
 
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
@@ -26,6 +27,7 @@ CLASS_RISK = {
 }
 COLORS = ['#2ecc71','#3498db','#e74c3c','#f39c12','#9b59b6']
 
+@spaces.GPU
 def build_model():
     m = models.Sequential([
         layers.Input(shape=(187, 1)),
@@ -49,6 +51,7 @@ def build_model():
 
 model = build_model()
 
+@spaces.GPU
 def predict_ecg(text_input):
     try:
         cleaned = text_input.replace(",", " ").replace("\n", " ").replace("\t", " ")