abhishtagatya
commited on
Commit
•
cb2350d
1
Parent(s):
801323e
Update README.md
Browse files
README.md
CHANGED
@@ -33,10 +33,10 @@ It achieves the following results on the evaluation set:
|
|
33 |
```python
|
34 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
35 |
|
36 |
-
config = AutoConfig.from_pretrained("abhishtagatya/
|
37 |
-
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/
|
38 |
|
39 |
-
model = Wav2Vec2ForSequenceClassification.from_pretrained("abhishtagatya/
|
40 |
|
41 |
# Your Logic Here
|
42 |
```
|
|
|
33 |
```python
|
34 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
35 |
|
36 |
+
config = AutoConfig.from_pretrained("abhishtagatya/wav2vec2-base-960h-itw-deepfake")
|
37 |
+
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/wav2vec2-base-960h-itw-deepfake")
|
38 |
|
39 |
+
model = Wav2Vec2ForSequenceClassification.from_pretrained("abhishtagatya/wav2vec2-base-960h-itw-deepfake", config=config).to(device)
|
40 |
|
41 |
# Your Logic Here
|
42 |
```
|