Update app.py

app.py

@@ -4,6 +4,7 @@ from sklearn.model_selection import train_test_split
 from sklearn.linear_model import LogisticRegression
 from joblib import dump, load
 import os
+from huggingface_hub import HfApi
 
 # Função para treinar e salvar o modelo
 def train_model():
@@ -25,6 +26,18 @@ def train_model():
     model_filename = os.path.join(diretorio, "model.pkl")
     dump(model, model_filename)
     
+    # Fazer upload do modelo para o Hugging Face
+    api = HfApi()
+    repo_id = "coan/botClaiton"  # Substitua pelo seu repositório
+
+    # Upload do modelo para o repositório Hugging Face
+    api.upload_file(
+        path_or_fileobj=model_filename,
+        path_in_repo="model.pkl",  # Nome do arquivo no repositório
+        repo_id=repo_id,
+        repo_type="model"
+    )
+    
     return f"Modelo treinado e salvo em: {model_filename}"
 
 # Função para carregar o modelo treinado e fazer previsões
@@ -32,7 +45,10 @@ def predict(input_data):
     # Carregar o modelo treinado
     model = load("/mnt/data/model.pkl")  # Caminho do modelo salvo
 
-    # Realizar a previsão (garantir que a entrada seja formatada corretamente)
+    # Garantir que a entrada seja um vetor (as características)
+    input_data = [float(i) for i in input_data.split()]
+    
+    # Realizar a previsão
     prediction = model.predict([input_data])  # input_data deve ser um vetor com as características
     return f"Previsão: {prediction[0]}"
 
@@ -43,21 +59,6 @@ train_iface = gr.Interface(
     outputs=["text"]  # Exibe uma mensagem de sucesso
 )
 
-from huggingface_hub import HfApi
-
-# Autenticação
-api = HfApi()
-repo_id = "coan/botClaiton"  # Substitua pelo seu repositório
-model_filename = "/mnt/data/model.pkl"  # Caminho do seu modelo
-
-# Upload do modelo para o repositório
-api.upload_file(
-    path_or_fileobj=model_filename,
-    path_in_repo="model.pkl",  # Nome do arquivo no repositório
-    repo_id=repo_id,
-    repo_type="model"
-)
-
 # Interface para previsões
 predict_iface = gr.Interface(
     fn=predict,                  # Função para chamar ao gerar previsão