arturevs commited on
Commit
12484e8
·
1 Parent(s): d0c0c50

fixing something

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +2 -2
.gitignore CHANGED
@@ -18,6 +18,7 @@ __pycache__/
18
 
19
  # Ignorar arquivos de dados temporários (opcional, dependendo do fluxo de trabalho)
20
  *.csv
 
21
 
22
  # Ignorar arquivos de log
23
  *.log
 
18
 
19
  # Ignorar arquivos de dados temporários (opcional, dependendo do fluxo de trabalho)
20
  *.csv
21
+ *.joblib
22
 
23
  # Ignorar arquivos de log
24
  *.log
app.py CHANGED
@@ -8,8 +8,8 @@ from huggingface_hub import hf_hub_download
8
  @st.cache_resource
9
  def load_model():
10
  # Baixar o modelo e o LabelEncoder do Hugging Face
11
- model_path = hf_hub_download(repo_id="arturevs/90AnimalClassification", filename="mlp_classifier.joblib")
12
- label_encoder_path = hf_hub_download(repo_id="arturevs/90AnimalClassification", filename="label_encoder.joblib")
13
 
14
  # Carregar o modelo e o LabelEncoder
15
  model = joblib.load(model_path)
 
8
  @st.cache_resource
9
  def load_model():
10
  # Baixar o modelo e o LabelEncoder do Hugging Face
11
+ model_path = hf_hub_download(repo_id="arturevs/90AnimalClassifier", filename="mlp_classifier.joblib")
12
+ label_encoder_path = hf_hub_download(repo_id="arturevs/90AnimalClassifier", filename="label_encoder.joblib")
13
 
14
  # Carregar o modelo e o LabelEncoder
15
  model = joblib.load(model_path)