Update app.py

app.py

@@ -1,6 +1,6 @@
 # Adapted from https://github.com/guoyww/AnimateDiff/blob/main/app.py
 
-# import spaces
+import spaces
 import os
 import json
 import torch
@@ -67,7 +67,7 @@ class AnimateController:
         
         
         
-    #@spaces.GPU        
+    @spaces.GPU        
     def preprocess(self, image=None):
 
         model_key = "CompVis/stable-diffusion-v1-4"
@@ -89,7 +89,7 @@ class AnimateController:
 
         T.ToPILImage()(recon_image[0]).save(os.path.join(save_path, f'recon.jpg'))
 
-   #@spaces.GPU    
+    @spaces.GPU    
     def generate(self,  file=None, audio=None, prompt=None, 
                  cfg_scale=5, image_path=None,
                  pnp_f_t=0.8, pnp_attn_t=0.8,):