search

enzoampil / tito-joker

A humorous AI that uses state-of-the-art deep learning to tell jokes
http://35.225.94.177:8501/
GNU General Public License v3.0
45 stars 5 forks source link

Deploy model v2 by updating streamlit app #21

Closed enzoampil closed 4 years ago

enzoampil commented 4 years ago

Closes issue #20

The additional changes in the source code were required:

  1. Drop down in the streamlit app to choose the model version to run
  2. Dynamically setting the model source dir and stop_token, depending on which model is used
  3. Apply stop token truncation at the ID level. Necessary since v2 uses a special token as it's eos tag