mookkanvas commited on
Commit
f085c59
·
1 Parent(s): adeace5

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+
4
+ text_generator = pipeline("text-generation", model="gpt2")
5
+
6
+ st.title("Hugging Face Transformer Text Generation App")
7
+
8
+ input_text = st.text_area("Enter text for generation")
9
+
10
+ if input_text:
11
+ generated_text = text_generator(input_text, max_length=100, num_return_sequences=1)[0]['generated_text']
12
+
13
+ st.subheader("Generated Text:")
14
+ st.write(generated_text)