ApolloPro7/Generate_Script

1import streamlit as st2import requests3 4st.title("LLaMA2 TV Script Generator")5prompt = st.text_area("Enter your prompt:")6 7if st.button("Generate"):8    with st.spinner("Generating..."):9        response = requests.post(10            "http://localhost:8000/generate",11            json={"prompt": prompt}12        )13 14        if response.ok:15            st.markdown("### Output")16            st.write(response.json()["generated_script"])17        else:18            st.error("Something went wrong.")19