from transformers import pipeline
import streamlit as st

# Erstelle ein prompt generator
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')

# Definiere die Streamlit-Oberfläche
st.title("Prompt Generator")
prompt = st.text_input("Gib einen Satzanfang ein:")
button = st.button("Generieren")

# Generiere neuen Text beim Klick auf den Button
if button:
    text = generator(prompt, max_length=30, do_sample=True, temperature=0.7)[0]['generated_text']
    st.write(text)