Light Anti-hacking update

app.py

@@ -84,6 +84,11 @@ def predict(model, text, melody, duration, dimension, topk, topp, temperature, c
     else:
         if MOVE_TO_CPU:
             MODEL.to('cuda')
+    
+    # prevent hacking
+    duration = min(duration, 720)
+    overlap =  min(overlap, 15)
+    # 
 
     output = None
     segment_duration = duration

audiocraft/utils/extend.py

@@ -18,7 +18,7 @@ INTERRUPTING = False
 def separate_audio_segments(audio, segment_duration=30, overlap=1):
     sr, audio_data = audio[0], audio[1]
 
-    total_samples = len(audio_data)
+    total_samples = min(len(audio_data), 25)
     segment_samples = sr * segment_duration
     overlap_samples = sr * overlap