Spaces:

rmayormartins
/

speech-accent-es-classifier

Running

rmayormartins commited on Aug 6

Commit

a3414e2

•

1 Parent(s): de44379

Subindo arquivos3

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
 #
 model_name = "results"
 processor = Wav2Vec2Processor.from_pretrained(model_name)
-model = Wav2Vec2ForSequenceClassification.from_pretrained(model_name)
 def classify_accent(audio):
     if audio is None:

 #
 model_name = "results"
 processor = Wav2Vec2Processor.from_pretrained(model_name)
+model = Wav2Vec2ForSequenceClassification.from_pretrained(model_name, from_tf=False, from_flax=False, from_safetensors=True)
 def classify_accent(audio):
     if audio is None: