abhishtagatya
/

wav2vec2-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

cb2350d

•

1 Parent(s): 801323e

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -33,10 +33,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 = Wav2Vec2ForSequenceClassification.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/wav2vec2-base-960h-itw-deepfake")
+  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/wav2vec2-base-960h-itw-deepfake")
+  model = Wav2Vec2ForSequenceClassification.from_pretrained("abhishtagatya/wav2vec2-base-960h-itw-deepfake", config=config).to(device)
   # Your Logic Here
 ```