import streamlit as st
from transformers import pipeline

@st.cache_resource
def load_model():
    return pipeline("text-generation", model="Arni1ntares/Winner")

pipe = load_model()

st.title("Winner Model Text & Code Generation")
st.write("Enter your prompt below and get fast generation!")

prompt = st.text_area("Prompt", height=150)
max_tokens = st.slider("Max tokens to generate", 50, 512, 128)

if st.button("Generate"):
    if prompt.strip():
        with st.spinner("Generating..."):
            outputs = pipe(prompt, max_new_tokens=max_tokens)
            generated = outputs[0]["generated_text"]
        st.subheader("Generated Output")
        st.code(generated)
    else:
        st.warning("Please enter a prompt to generate text.")