Spaces:
Runtime error
CycleGAN Gradio Web App deployed on Heroku
This folder contains code to deploy a simple Horse-to-Zebra model onto Heroku for free.
An example web app is running at: http://upit-cyclegan.herokuapp.com/
Note: it takes a couple minutes for the web app to load and a few seconds to produce a result. Upgrading to paid web apps (dynos) may alleviate some of these issues.
How to deploy:
- Install the Heroku CLI
- Clone this repository:
git clone https://github.com/tmabraham/UPIT.git
- Initialize a Heroku web app from the command line (in the UPIT repository folder):
heroku create
- Deploy the
web_app
folder
git subtree push --prefix examples/web_app heroku master
And that's it! Heroku will process the code, compress it and deploy it, providing you with a link you can visit at any time!
This deployment was based on this guide.
File descriptions:
cyclegan_inference.ipynb
- This is a notebook with all the code to run a Gradio web app demo. Running this notebook will yield in a temporary temporary web app that is not persistent (runs for 6 hours). The gradio.app
link is provided after running the code. Also note that this code is set up to run on the GPU. This can be easily changed by changing the map_location
in torch.load
.
cyclegan_inference.py
- This is a Python file version of cyclegan_inference.ipynb
that runs on the CPU and takes in 256x256 images (due to low memory for free Heroku dyno). This is what is used for the Heroku deployment.
generator.pth
- My model file saved with the UPIT export_generator
function. You can replace with your own generator.
Procfile
,requirements.txt
,runtime.txt
,setup.sh
- These files are required for Heroku deployment.