Update app.py

app.py

@@ -116,17 +116,9 @@ def generate_midi(temperature, top_k):
             if sequence:
                 midi_events.append(sequence)
 
-    round_bars = []
-    
-    for sequence in midi_events:
-        filtered_sequence = []
-        for event in sequence:
-            if event['start'] < 768 and event['end'] <= 768:
-                filtered_sequence.append(event)
-        if filtered_sequence:
-            round_bars.append(filtered_sequence)
-
-    midi_events = round_bars
+    if midi_events:
+        midi_events = max(midi_events, key=len)
+        midi_events = [midi_events]
 
     return midi_events
 
@@ -238,7 +230,9 @@ iface = gr.Interface(
         gr.Audio(label="Generated Audio", type="filepath")
     ],
     title="<h1 style='font-weight: bold; text-align: center;'>nanoMPC - AI Midi Drum Sequencer</h1>",
-    description="<p style='text-align:center;'>nanoMPC is a tiny transformer model that generates MIDI drum beats inspired by Lo-Fi, Boom Bap and other styles of Hip Hop.</p>"
+    description="<p style='text-align:center;'>nanoMPC is a tiny transformer model that generates MIDI drum beats inspired by Lo-Fi, Boom Bap and other styles of Hip Hop.</p>",
+    theme="soft",
+    allow_flagging="never",
 )
 
 iface.launch()