Update app.py

app.py

@@ -5,6 +5,8 @@ import gradio as gr
 import spaces
 import os
 import uuid
+import shutil
+import gzip
 
 # Importing the model-related functions
 from stable_audio_tools import get_pretrained_model
@@ -16,6 +18,14 @@ def load_model():
     model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
     print("Model loaded successfully.")
     return model, model_config
+    
+def compress_file(file_path):
+    compressed_file_path = file_path + '.gz'
+    with open(file_path, 'rb') as f_in:
+        with gzip.open(compressed_file_path, 'wb') as f_out:
+            shutil.copyfileobj(f_in, f_out)
+    return compressed_file_path
+
 
 # Function to set up, generate, and process the audio
 @spaces.GPU(duration=60)  # Allocate GPU only when this function is called
@@ -77,10 +87,13 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
 
     # Save to file
     torchaudio.save(unique_filename, output, sample_rate)
-    print(f"Audio saved: {unique_filename}")
+    print(f"Audio saved: {unique_filename}")]
+    
+    compressed_filename = compress_file(unique_filename)
+    return compressed_filename
 
-    # Return the path to the generated audio file
-    return unique_filename
+    # # Return the path to the generated audio file
+    # return unique_filename
 
 
 DESCRIPTION = "Welcome to Raptor APIs"