Marcos12886 commited on
Commit
2ca1b49
1 Parent(s): ace06e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -6,15 +6,15 @@ import numpy as np
6
  from model import SAMPLING_RATE, FEATURE_EXTRACTOR
7
 
8
  token = os.getenv("HF_TOKEN")
9
- modelo = "mixed-data"
10
- # modelo = "cry-detector"
11
  pipe = pipeline(
12
  "audio-classification",
13
  model=f"A-POR-LOS-8000/distilhubert-finetuned-{modelo}",
14
  use_auth_token=token
15
  )
16
- # client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct", token=token)
17
- client = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407", token=token)
18
 
19
  def respond(
20
  message,
 
6
  from model import SAMPLING_RATE, FEATURE_EXTRACTOR
7
 
8
  token = os.getenv("HF_TOKEN")
9
+ # modelo = "mixed-data"
10
+ modelo = "cry-detector"
11
  pipe = pipeline(
12
  "audio-classification",
13
  model=f"A-POR-LOS-8000/distilhubert-finetuned-{modelo}",
14
  use_auth_token=token
15
  )
16
+ client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct", token=token)
17
+ # client = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407", token=token)
18
 
19
  def respond(
20
  message,