import spaces , add decorator

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.mp4 filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import py3Dmol
 import io
@@ -309,6 +310,7 @@ def prediction_visualization(pdb_file, num_runs: int, noise_level: float, num_fr
     view_data = visualize_after_pred(protein, best_aligned[1])
     return view_data, f"Best cRMSD: {best_aligned[0]:.4f}"
 
+@spaces.GPU
 def run_prediction(pdb_file, num_runs, noise_level, num_frames, temperature, num_steps, progress=gr.Progress()):
     try:
         if pdb_file is None:

demovideo/demo.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa79153abc111e4686e121dca7fb9a8ca7dbde9b2e2555b5a5d117ad2452edc5
+size 11042989