README / app.py
Kirenz's picture
Create app.py
45ff229 verified
from transformers import pipeline
import streamlit as st
def create_product_description_app():
st.title("KI-gestützter Produktbeschreibungsgenerator")
# Interface erstellen
product_name = st.text_input("Produktname:")
key_features = st.text_area("Hauptmerkmale (durch Kommas getrennt):")
target_audience = st.text_input("Zielgruppe:")
if st.button("Beschreibung generieren"):
# Prompt zusammenbauen
prompt = f"""
Produkt: {product_name}
Merkmale: {key_features}
Zielgruppe: {target_audience}
Erstelle eine verkaufsfördernde Produktbeschreibung.
"""
# Generator initialisieren
generator = pipeline('text-generation',
model='gpt2')
# Beschreibung generieren
description = generator(prompt,
max_length=200,
num_return_sequences=1)[0]['generated_text']
st.write("Generierte Beschreibung:")
st.write(description)
if __name__ == "__main__":
create_product_description_app()