Spaces:
Sleeping
Sleeping
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() |