Spaces:
Running
Running
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) |