Jabrain commited on
Commit
2d5548b
·
1 Parent(s): fb506ba

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -0
app.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import streamlit as st
3
+ from transformers import T5Tokenizer, T5ForConditionalGeneration
4
+
5
+ st.title("Text-to-Text AI Model")
6
+
7
+ st.write("Enter your prompt below:")
8
+
9
+ prompt = st.text_input("Prompt")
10
+
11
+ if prompt:
12
+ tokenizer = T5Tokenizer.from_pretrained('t5-small')
13
+ model = T5ForConditionalGeneration.from_pretrained('t5-small')
14
+
15
+ input_ids = tokenizer.encode(prompt, return_tensors='pt')
16
+ output = model.generate(input_ids)
17
+ output_text = tokenizer.decode(output[0], skip_special_tokens=True)
18
+
19
+ st.write("Output:")
20
+ st.write(output_text)