Este repositório contém código para o protótipo de aplicativo desenvolvido como parte do meu projeto de pesquisa em MSc Data Science na Nottingham Trent University. O repositório contém dois aplicativos principais; um aplicativo baseado em nuvem e um aplicativo no dispositivo.
AppFastAPI: ArtBrain desenvolvido usando FastAPI.
AppTFJS: ArtBrain desenvolvido usando TensorFlowJS.
Python 3.10+
PIP
GPU compatível com CUDA com pelo menos 10 GB de VRAM
A instalação do CUDA também pode ser necessária.
Execute o seguinte comando para instalar as dependências. Recomendado para uso e ambiente virtual.
pip install -r requirements.txt
Coloque os arquivos do modelo em suas respectivas pastas. Entre em contato com o autor para obter os modelos treinados.
Para executar o aplicativo FastAPI, execute os seguintes comandos no terminal.
cd AppFastAPI python3 main.py
Após a execução desta linha você pode visitar seu localhost para utilizar a aplicação. Alternativamente, você pode usar o Docker para hospedar o aplicativo usando o Dockerfile fornecido.
Para usar o aplicativo TensorFlowJS, basta abrir o arquivo js_art_home.html em um navegador da web. Certifique-se de ter as pastas do modelo no mesmo diretório do repositório.
Ravidu Suien Rammuni Silva