abhishtagatya
/

wavlm-base-960h-itw-deepfake

Audio Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

abhishtagatya commited on Jan 31

Commit

1a9795f

•

1 Parent(s): d8046c9

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -32,10 +32,10 @@ It achieves the following results on the evaluation set:
 ```python
   device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-  config = AutoConfig.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
-  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
-  model = WavLMForSequenceClassification.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake", config=config).to(device)
   # Your Logic Here
 ```

 ```python
   device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+  config = AutoConfig.from_pretrained("abhishtagatya/wavlm-base-960h-itw-deepfake")
+  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/wavlm-base-960h-itw-deepfake")
+  model = WavLMForSequenceClassification.from_pretrained("abhishtagatya/wavlm-base-960h-itw-deepfake", config=config).to(device)
   # Your Logic Here
 ```