einfach.prompt / app.py
AIlexDev's picture
Update app.py
2a6f27f
raw
history blame
569 Bytes
import gradio as gr
from transformers import pipeline
generator = pipeline("text2text-generation", model="merve/chatgpt-prompts-bart-long")
def generate_prompt(prompt):
output = generator(prompt, max_length=250)
return output[0]['generated_text']
iface = gr.Interface(fn=generate_prompt, inputs="text", outputs="text", title="💯 Einfach.Prompt Generator 💯 made with ",
description="Wunsch prompt eingeben z.b. Web Designer und einen prompt für Chat.GPT erstellen lassen",
theme="huggingface")
iface.launch()