import streamlit as st
from transformers import pipeline

task = st.selectbox("Choose a task", ["Text generation", "Summarization"])

if task == "Text generation":
    model = pipeline("text-generation")
else:
    model = pipeline("summarization")
    
prompt = st.text_area("Enter a prompt")

if prompt:
    output = model(prompt)[0]["generated_text"]
    st.write(output)

'''This sets up a simple interface with a dropdown to select either text generation or summarization,
a text area for input, and displays the generated output'''