File size: 556 Bytes
2d5548b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import streamlit as st
from transformers import T5Tokenizer, T5ForConditionalGeneration
st.title("Text-to-Text AI Model")
st.write("Enter your prompt below:")
prompt = st.text_input("Prompt")
if prompt:
tokenizer = T5Tokenizer.from_pretrained('t5-small')
model = T5ForConditionalGeneration.from_pretrained('t5-small')
input_ids = tokenizer.encode(prompt, return_tensors='pt')
output = model.generate(input_ids)
output_text = tokenizer.decode(output[0], skip_special_tokens=True)
st.write("Output:")
st.write(output_text)
|