FROM tuvbui/torchcpu:torch111
ADD cldm ./cldm
ADD flae ./flae
ADD ldm ./ldm
ADD tools ./tools
ADD pages ./pages
ADD Embed_Secret.py .

EXPOSE 7860
CMD streamlit run Embed_Secret.py --server.enableXsrfProtection=false --server.port 7860 -- --weight https://kahlan.cvssp.org/data/Flickr25K/tubui/stega/unet100b_croprs/epoch=000070-step=000219999.ckpt --config https://kahlan.cvssp.org/data/Flickr25K/tubui/stega/unet100b_croprs/-project.yaml