St4 / app.py
ali121300's picture
Update app.py
3159508 verified
raw
history blame contribute delete
240 Bytes
import streamlit as st
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-large")
model = AutoModelForSeq2SeqLM.from_pretrained("google-t5/t5-large")