ZeroGPU test

app.py

@@ -1,6 +1,7 @@
 import re, os
 from pathlib import Path
 import gradio as gr
+import spaces
 import torch
 
 from evodiff.pretrained import OA_DM_38M, D3PM_UNIFORM_38M, MSA_OA_DM_MAXSUB
@@ -8,7 +9,7 @@ from evodiff.generate import generate_oaardm, generate_d3pm
 from evodiff.generate_msa import generate_query_oadm_msa_simple
 from evodiff.conditional_generation import inpaint_simple, generate_scaffold
 
-
+@spaces.GPU
 def get_device():
     if torch.cuda.is_available():
         return "cuda"