src/streamlit_app.py

import streamlit as st
from transformers import pipeline

st.set_page_config(page_title="AI Quote Generator", page_icon="✨")
st.title("🎭 AI Quote Generator")

@st.cache_resource
def load_model():
    return pipeline('text-generation', model='nandinib1999/quote-generator')

generator = load_model()

topic = st.text_input("Enter a topic:", "Success")
temp = st.slider("Creativity", 0.5, 1.2, 0.85)

if st.button("Generate Quote"):
    result = generator(f"Quote about {topic}:", max_length=70, temperature=temp, do_sample=True)
    quote = result[0]['generated_text']
    quote = quote.replace(f"Quote about {topic}:", '').strip()
    st.markdown(f"> *\"{quote}\"*")