init!

demo.py

@@ -0,0 +1,15 @@
+import streamlit as st
+from transformers import pipeline
+
+# Load the text generation pipeline
+generator = pipeline("text-generation", model="gpt2")
+
+# Streamlit UI
+st.title("Text Generation with Hugging Face")
+
+prompt = st.text_input("Enter your text prompt:", "Once upon a time,")
+
+if st.button("Generate Text"):
+    # Generate text based on the prompt
+    generated_text = generator(prompt, max_length=100, do_sample=True)[0]["generated_text"]
+    st.write(generated_text)