Check / app.py
Alfasign's picture
Update app.py
4dd0859
raw
history blame
493 Bytes
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)