tykiww's picture
Update server.py
cf6451a verified
raw
history blame contribute delete
255 Bytes
def submit_weights(model, repository, model_out_name, token):
"""submits model to repository"""
repo = repository + '/' + model_out_name
model.push_to_hub(repo, token = token)
tokenizer.push_to_hub(repo, token = token)
return 0