Spaces:
Sleeping
Sleeping
Marcos12886
commited on
Commit
·
1565b0a
1
Parent(s):
53f6532
Acceder modelo hf
Browse files
app.py
CHANGED
@@ -9,13 +9,13 @@ token = os.getenv("HF_TOKEN")
|
|
9 |
client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct", token=token)
|
10 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
11 |
model_class, id2label_class = predict_params(
|
12 |
-
model_path="distilhubert-finetuned-mixed-data",
|
13 |
dataset_path="data/mixed_data",
|
14 |
filter_white_noise=True,
|
15 |
undersample_normal=True
|
16 |
)
|
17 |
model_mon, id2label_mon = predict_params(
|
18 |
-
model_path="distilhubert-finetuned-cry-detector",
|
19 |
dataset_path="data/baby_cry_detection",
|
20 |
filter_white_noise=False,
|
21 |
undersample_normal=False
|
|
|
9 |
client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct", token=token)
|
10 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
11 |
model_class, id2label_class = predict_params(
|
12 |
+
model_path="A-POR-LOS-8000/distilhubert-finetuned-mixed-data",
|
13 |
dataset_path="data/mixed_data",
|
14 |
filter_white_noise=True,
|
15 |
undersample_normal=True
|
16 |
)
|
17 |
model_mon, id2label_mon = predict_params(
|
18 |
+
model_path="A-POR-LOS-8000/distilhubert-finetuned-cry-detector",
|
19 |
dataset_path="data/baby_cry_detection",
|
20 |
filter_white_noise=False,
|
21 |
undersample_normal=False
|