tlemagueresse
commited on
Commit
·
54d215b
1
Parent(s):
672b0df
Fix import
Browse files- README.md +1 -1
- __init__.py +1 -0
README.md
CHANGED
|
@@ -65,7 +65,7 @@ FastModelHuggingFace = model_module.FastModelHuggingFace
|
|
| 65 |
fast_model = FastModelHuggingFace.from_pretrained(repo_id)
|
| 66 |
|
| 67 |
# Perform predictions
|
| 68 |
-
result =
|
| 69 |
map_labels = {0: "chainsaw", 1: "environment"}
|
| 70 |
print(f"Prediction Result: {map_labels[result[0]]}")
|
| 71 |
```
|
|
|
|
| 65 |
fast_model = FastModelHuggingFace.from_pretrained(repo_id)
|
| 66 |
|
| 67 |
# Perform predictions
|
| 68 |
+
result = fast_model.predict("path/to/audio.wav")
|
| 69 |
map_labels = {0: "chainsaw", 1: "environment"}
|
| 70 |
print(f"Prediction Result: {map_labels[result[0]]}")
|
| 71 |
```
|
__init__.py
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
from model import *
|