peg requirements

zero-gpu adjustments

app.py

@@ -4,6 +4,7 @@ import shutil
 import zipfile
 from io import BytesIO
 
+import spaces
 import gradio as gr
 import imageio as imageio
 import numpy as np
@@ -45,6 +46,7 @@ default_bas_frame_near = 1
 default_bas_frame_far = 1
 
 
+@spaces.GPU
 def process_image(
     pipe,
     path_input,
@@ -87,6 +89,7 @@ def process_image(
     )
 
 
+@spaces.GPU
 def process_video(
     pipe,
     path_input,
@@ -184,6 +187,7 @@ def process_video(
     )
 
 
+@spaces.GPU
 def process_bas(
     pipe,
     path_input,

requirements.txt

@@ -6,10 +6,10 @@ imageio
 imageio-ffmpeg
 Pillow
 
-accelerate>=0.22.0
+accelerate==0.28.0
 diffusers==0.27.2
 matplotlib==3.8.2
 scipy==1.11.4
 torch==2.0.1
-transformers>=4.32.1
-xformers>=0.0.21
+transformers==4.39.1
+xformers==0.0.25