dominikmn / one-million-posts

Assisting newspaper moderators with machine learning.
MIT License
2 stars 1 forks source link

Create gbert prediction function for the dashboard #126

Closed dominikmn closed 3 years ago

dominikmn commented 3 years ago

Create a function that loads a PyTorch model an predicts the classes for one particular string.