server.py File Reference

Namespaces
namespace	server

Functions
	server.generate ()

Variables
	server.app = Flask(__name__)
int	server.PLUGIN_PARAM_COUNT = 9
str	server.MODEL_PATH = "../saved_models/my_plugin_ai.pth"
list	server.PARAM_KEYS
	server.device = torch.device('cpu')
	server.model = TextToParams(num_plugin_parameters=PLUGIN_PARAM_COUNT)
	server.tokenizer = AutoTokenizer.from_pretrained("prajjwal1/bert-tiny")
	server.port