Updates to Auto classes
Browse files
README.md
CHANGED
@@ -135,8 +135,8 @@ if HfFolder.get_token() is None:
|
|
135 |
|
136 |
|
137 |
model_id = "audiblehealthai/hear-pytorch"
|
138 |
-
model =
|
139 |
-
fe =
|
140 |
model_id
|
141 |
)
|
142 |
|
@@ -215,8 +215,8 @@ if HfFolder.get_token() is None:
|
|
215 |
|
216 |
|
217 |
model_id = "audiblehealthai/hear-pytorch"
|
218 |
-
classifier =
|
219 |
-
fe =
|
220 |
model_id
|
221 |
)
|
222 |
|
|
|
135 |
|
136 |
|
137 |
model_id = "audiblehealthai/hear-pytorch"
|
138 |
+
model = AutoModel.from_pretrained(model_id)
|
139 |
+
fe = AutoFeatureExtractor.from_pretrained(
|
140 |
model_id
|
141 |
)
|
142 |
|
|
|
215 |
|
216 |
|
217 |
model_id = "audiblehealthai/hear-pytorch"
|
218 |
+
classifier = AutoModelForAudioClassification.from_pretrained(model_id)
|
219 |
+
fe = AutoFeatureExtractor.from_pretrained(
|
220 |
model_id
|
221 |
)
|
222 |
|