Update app.py

app.py

@@ -24,13 +24,13 @@ def get_training_info(audio_file, pretrain):
             sample_rate = audio.getframerate()
 
         training_info = {
-            (0, 2): (150, f'{pretrain}'),
-            (2, 3): (200, f'{pretrain}'),
-            (3, 5): (250, f'{pretrain}'),
-            (5, 10): (300, f'{pretrain}'),
-            (10, 25): (500, f'{pretrain}'),
-            (25, 45): (700, f'{pretrain}'),
-            (45, 60): (1000, f'{pretrain}')
+            (0, 2): (150, 'OV2'),
+            (2, 3): (200, 'OV2'),
+            (3, 5): (250, 'OV2'),
+            (5, 10): (300, 'OV2'),
+            (10, 25): (500, 'Normal'),
+            (25, 45): (700, 'Normal'),
+            (45, 60): (1000, 'Normal')
         }
 
         for (min_duration, max_duration), (epochs, pretrain) in training_info.items():