Docto-Bot / app.py
kazimsayed's picture
Update app.py
298fce5
raw
history blame
232 Bytes
import random
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("docto/Docto-Bot")
model = AutoModelForCausalLM.from_pretrained("docto/Docto-Bot")
special_token = '<|endoftext|>'