Update app.py

app.py

@@ -4,13 +4,14 @@ from in2in.utils.plot import plot_3d_motion
 from in2in.utils.paramUtil import HML_KINEMATIC_CHAIN
 from transformers import AutoModel        
 
+@gpu
 def generate(textI, texti1, texti2):
     preds = model(textI, texti1, texti2)
     plot_3d_motion("result.mp4", HML_KINEMATIC_CHAIN, preds, title="", fps=30)
     return "result.mp4"
 
 model = AutoModel.from_pretrained("pabloruizponce/in2IN", trust_remote_code=True)
-#model.to("cuda")
+model.to("cuda")
 demo = gr.Interface(fn=generate, 
                     inputs=[gr.Text(label="Interaction Description"),
                             gr.Text(label="Individual1 Description"),