Spaces:
Build error
Build error
File size: 1,372 Bytes
7c8a29e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
#!/usr/bin/env python3
"""
Arquivo principal para deploy no Hugging Face Spaces
LoRA Trainer - Ferramenta de Treinamento de Personagens Consistentes
"""
import os
import sys
# Adicionar o diretório src ao path
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
# Importar a aplicação Flask
from main import app
# Configurações específicas para Hugging Face Spaces
app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024 # 500MB
app.config['UPLOAD_FOLDER'] = '/tmp/lora_uploads'
app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY', 'lora-trainer-secret-key-2024')
# Configurar CORS para Hugging Face
from flask_cors import CORS
try:
CORS(app, origins=["*"])
except ImportError:
# Se flask-cors não estiver disponível, configurar manualmente
@app.after_request
def after_request(response):
response.headers.add('Access-Control-Allow-Origin', '*')
response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')
return response
if __name__ == "__main__":
# Porta padrão do Hugging Face Spaces
port = int(os.environ.get("PORT", 7860))
# Executar aplicação
app.run(
host="0.0.0.0",
port=port,
debug=False # Desabilitar debug em produção
)
|