Update app.py

app.py

@@ -9,8 +9,8 @@ from app.model import predict
 from app.preprocess import preprocess_audio
 
 # Load Hugging Face datasets directly
-audio_ds = load_dataset("AIOmarRehan/General_Audio_Dataset", split="train")
-image_ds = load_dataset("AIOmarRehan/Mel_Spectrogram_Images_for_Audio_Classification", split="train")
+audio_ds = load_dataset("AIOmarRehan/General_Audio_Dataset")
+image_ds = load_dataset("AIOmarRehan/Mel_Spectrogram_Images_for_Audio_Classification")
 
 # Helper function to safely load images
 def safe_load_image(img):