trial2 / app.py
alaa-ahmed14's picture
Create app.py
31cc367 verified
raw
history blame
210 Bytes
from transformers import AutoModelForCausalLM, AutoTokenizer
import os
tokenizer = AutoTokenizer.from_pretrained("matsant01/STEMerald-2b")
model = AutoModelForCausalLM.from_pretrained("matsant01/STEMerald-2b")