Fixes for ZeroGPU compatibility

app.py

@@ -3,6 +3,7 @@ from engine import DescribeVideo, GenerateAudio
 from moviepy.editor import VideoFileClip, AudioFileClip, CompositeAudioClip
 from moviepy.audio.fx.volumex import volumex
 import shutil, tempfile, os
+import spaces
 
 # Maps for model selection based on user input
 video_model_map = {
@@ -115,6 +116,7 @@ def generate_video_description(video_descriptor, google_api_key, toggle_advanced
         raise gr.Error("Exception raised: ", e)
 
 # Function to generate music based on the video description    
+@spaces.GPU
 def generate_music(music_generator, music_prompt, num_samples):
     global video_duration, audio_paths, session_dir
     try:

environment.yml

@@ -161,7 +161,7 @@ dependencies:
       - prompt-toolkit==3.0.47
       - proto-plus==1.24.0
       - protobuf==4.25.3
-      - psutil==6.0.0
+      - psutil==5.9.8
       - ptyprocess==0.7.0
       - pure-eval==0.2.3
       - pyarrow==16.1.0
@@ -201,6 +201,7 @@ dependencies:
       - soundfile==0.12.1
       - soupsieve==2.5
       - soxr==0.3.7
+      - spaces==0.29.3
       - spacy==3.7.5
       - spacy-legacy==3.0.12
       - spacy-loggers==1.0.5

requirements.txt

@@ -137,7 +137,7 @@ proglog==0.1.10
 prompt_toolkit==3.0.47
 proto-plus==1.24.0
 protobuf==4.25.3
-psutil==6.0.0
+psutil==5.9.8
 ptyprocess==0.7.0
 pure_eval==0.2.3
 pyarrow==16.1.0
@@ -177,6 +177,7 @@ sniffio==1.3.1
 soundfile==0.12.1
 soupsieve==2.5
 soxr==0.3.7
+spaces==0.29.3
 spacy==3.7.5
 spacy-legacy==3.0.12
 spacy-loggers==1.0.5