Update app.py

app.py

@@ -27,10 +27,18 @@ def loading_sound_file(sound_file, sr=22050, duration=10):
     #     y = lib.util.fix_length(X, input_length) 
     # extract normalized mfcc feature from data
     
-    ## pad audio to same duration 
+    # ## pad audio to same duration 
+    # if round(dur) < duration:
+    #     X = lib.util.fix_length(X, input_length)
+
+    # Pad or truncate audio file to the same duration
     if round(dur) < duration:
-        X = lib.util.fix_length(X, input_length)
-    
+        pad_amount = input_length - len(X)
+        X = np.pad(X, (0, pad_amount), mode='constant')
+    elif round(dur) > duration:
+        X = X[:input_length]
+
+        
     mfccs = np.mean(lib.feature.mfcc(y=X, sr=sr, n_mfcc=25).T,axis=0)
 
     ## Reshape to match the model's input shape
@@ -68,7 +76,7 @@ def heart_signal_classification(data):
 title = "Heart Signal Classification App"   
 Input = gr.Audio(sources=["upload"], type="filepath")
 Output1 = gr.Textbox(label="Type Of Heart Signal")
-description = "Type Of Signal: Artifact, Murmur, Normal"
+description = "Type Of Signal: Artifact, Murmur, Normal, Extrastole, Extrahls"
 iface = gr.Interface(fn=heart_signal_classification, inputs=Input, outputs=Output1, title=title, description=description)
 
 iface.launch(inline=False)