abhishtagatya
commited on
Commit
•
965d306
1
Parent(s):
54318dd
Update README.md
Browse files
README.md
CHANGED
@@ -38,7 +38,7 @@ It achieves the following results on the evaluation set:
|
|
38 |
config = AutoConfig.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
|
39 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
|
40 |
|
41 |
-
model = HubertForSequenceClassification.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake", config=config
|
42 |
|
43 |
# Your Logic Here
|
44 |
```
|
|
|
38 |
config = AutoConfig.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
|
39 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
|
40 |
|
41 |
+
model = HubertForSequenceClassification.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake", config=config).to(device)
|
42 |
|
43 |
# Your Logic Here
|
44 |
```
|