Spaces:

A-POR-LOS-8000
/

CHATBOT

Sleeping

Marcos12886 commited on Sep 10

Commit

f72ba5a

•

1 Parent(s): d0064d4

problema con ruta

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,8 +8,8 @@ from interfaz import estilo, my_theme
 token = os.getenv("HF_TOKEN")
 client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct", token=token)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_class, id2label_class = predict_params(model_path="A-POR-LOS-8000/distilhubert-finetuned-mixed-data", dataset_path="data/mixed_data", filter_white_noise=True)
-model_mon, id2label_mon = predict_params(model_path="A-POR-LOS-8000/distilhubert-finetuned-cry-detector", dataset_path="data/baby_cry_detection", filter_white_noise=False)
 def call(audiopath, model, dataset_path, filter_white_noise):
     model.to(device)

 token = os.getenv("HF_TOKEN")
 client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct", token=token)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_class, id2label_class = predict_params(model_path="A-POR-LOS-8000/distilhubert-finetuned-mixed-data", dataset_path="A-POR-LOS-8000/data/mixed_data", filter_white_noise=True)
+model_mon, id2label_mon = predict_params(model_path="A-POR-LOS-8000/distilhubert-finetuned-cry-detector", dataset_path="A-POR-LOS-8000/data/baby_cry_detection", filter_white_noise=False)
 def call(audiopath, model, dataset_path, filter_white_noise):
     model.to(device)