Update app.py

app.py

@@ -26,10 +26,17 @@ def train_model():
     model_filename = os.path.join(diretorio, "model.pkl")
     dump(model, model_filename)
     
-    # Fazer upload do modelo para o Hugging Face
+    # Criar repositório no Hugging Face se não existir
     api = HfApi()
     repo_id = "coan/botClaiton"  # Substitua pelo seu repositório
 
+    # Verificar se o repositório já existe
+    try:
+        api.repo_info(repo_id)
+    except:
+        # Criar o repositório caso não exista
+        api.create_repo(repo_id, repo_type="model")
+
     # Upload do modelo para o repositório Hugging Face
     api.upload_file(
         path_or_fileobj=model_filename,