feat: add gpu support

app.py

@@ -6,6 +6,7 @@ from pathlib import Path
 import numpy as np
 import gradio as gr
 import asyncio
+import spaces
 
 # Set matplotlib to use non-interactive backend before importing pyplot
 # This is required for Gradio which runs on worker threads
@@ -383,6 +384,7 @@ def reset_on_species_change(species: str):
 # -----------------------------
 # Predict
 # -----------------------------
+@spaces.GPU
 def predict(
     seq: str,
     species: str,