Update app.py

app.py

@@ -1,11 +1,15 @@
+import os
 import torch
 import torchaudio
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Tokenizer
 
+# Set the Hugging Face API token
+os.environ["HUGGINGFACE_TOKEN"] = "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"
+
 # Load the pre-trained model and tokenizer
-model_name = "BrendaTellez/SoundClassificationCNNRNN"
-model = Wav2Vec2ForCTC.from_pretrained(model_name)
-tokenizer = Wav2Vec2Tokenizer.from_pretrained(model_name)
+model_name = "BrendaTellez/SoundClassification"
+model = Wav2Vec2ForCTC.from_pretrained(model_name, use_auth_token=True)
+tokenizer = Wav2Vec2Tokenizer.from_pretrained(model_name, use_auth_token=True)
 
 # Get the audio file from the user
 file_path = input("Enter the path to the audio file: ")
@@ -29,3 +33,4 @@ predicted_class_id = torch.argmax(logits, dim=-1)
 predicted_class_label = tokenizer.decode(predicted_class_id[0])
 
 print(f"The audio file is classified as: {predicted_class_label}")
+