Added GPU duration

app.py

@@ -168,7 +168,7 @@ def convert_midi_to_wav(midi_path):
         print(f"Error converting MIDI to WAV: {str(e)}")
         return None
 
-@spaces.GPU # Set duration to 60 seconds for music generation
+@spaces.GPU(duration=120) # Set duration to 60 seconds for music generation
 def generate_music(image, conditioning_type, gen_len, temperature, top_p, min_instruments):
     """Generate music from input image"""
     model = get_model(conditioning_type)

aria/aria.py

@@ -15,7 +15,7 @@ sys.path.append(MIDI_EMOTION_PATH)
 class ARIA:
     """ARIA model that generates music from images based on emotional content."""
     
-    @spaces.GPU  # Model loading should be quick
+    @spaces.GPU(duration=20) # Model loading should be quick
     def __init__(
         self,
         image_model_checkpoint: str,
@@ -60,7 +60,7 @@ class ARIA:
         self.midi_model.load_state_dict(torch.load(model_fp, map_location=self.device, weights_only=True))
         self.midi_model.eval()
     
-    @spaces.GPU
+    @spaces.GPU(duration=120)
     @torch.inference_mode()  # More efficient than no_grad for inference
     def generate(
         self,