Funnybubble / app.py
Funybubble's picture
Update app.py
d3f153c verified
raw
history blame
546 Bytes
from transformers import AutoModel, AutoTokenizer
# Replace 'Funybubble/replit-code-instruct-quanatized' with the actual model name
model_name = 'Funybubble/replit-code-instruct-quanatized'
# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModel.from_pretrained(model_name)
# Save the tokenizer and model to the same directory as app.py
save_directory = './'
tokenizer.save_pretrained(save_directory)
model.save_pretrained(save_directory)
# Rest of your code using the tokenizer and model
# ...