Create app.py

app.py

@@ -0,0 +1,35 @@
+# app.py
+import streamlit as st
+from transformers import pipeline
+
+# Page config
+st.set_page_config(page_title="AI Text Generator", page_icon="🤖")
+
+st.title("🤖 AI Text Generator")
+st.write("Enter a prompt and get AI-generated text!")
+
+# Prompt input
+prompt = st.text_area("Your prompt here:", height=150)
+
+# Optional sliders for advanced users
+st.sidebar.header("Generation Settings")
+max_length = st.sidebar.slider("Max Length", min_value=50, max_value=500, value=200, step=10)
+temperature = st.sidebar.slider("Temperature", min_value=0.1, max_value=1.0, value=0.7, step=0.1)
+top_k = st.sidebar.slider("Top-k sampling", min_value=10, max_value=100, value=50, step=5)
+
+# Generate button
+if st.button("Generate"):
+    if prompt.strip() == "":
+        st.warning("Please enter a prompt!")
+    else:
+        with st.spinner("Generating text..."):
+            generator = pipeline("text-generation", model="gpt2")  # You can change to any HF model
+            result = generator(
+                prompt,
+                max_length=max_length,
+                do_sample=True,
+                temperature=temperature,
+                top_k=top_k
+            )
+            st.subheader("Generated Text")
+            st.write(result[0]['generated_text'])