abhishtagatya commited on
Commit
801323e
1 Parent(s): e75371b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -36,7 +36,7 @@ It achieves the following results on the evaluation set:
36
  config = AutoConfig.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
37
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
38
 
39
- model = HubertForSequenceClassification.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake", config=config).to(device)
40
 
41
  # Your Logic Here
42
  ```
 
36
  config = AutoConfig.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
37
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
38
 
39
+ model = Wav2Vec2ForSequenceClassification.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake", config=config).to(device)
40
 
41
  # Your Logic Here
42
  ```