import streamlit as st
from transformers import pipeline

# Load the text generation pipeline
generator = pipeline("text-generation", model="google/gemma-7b", tokenizer="google/gemma-7b")

# Streamlit app title and description
st.title("Gemma Text Generation App")
st.write("This app generates text based on the input prompt using the Gemma model.")

# Text input for user prompt
prompt = st.text_input("Enter your prompt:", "Once upon a time,")

# Generate button to trigger text generation
if st.button("Generate Text"):
    # Generate text based on the user prompt
    generated_text = generator(prompt, max_length=100)[0]['generated_text']
    # Display the generated text
    st.write("Generated Text:")
    st.write(generated_text)