EinfachPrompt / app.py
TogetherAI's picture
Update app.py
c4b8860
raw
history blame
1.19 kB
import gradio as gr
from transformers import pipeline
# Titel, Beschreibung und Artikel für die Schnittstelle
title = "Prompt Dich weiter mit Einfach.Prompt"
description = """
<p>
<center>
</center>
</p>
"""
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1907.06616' target='_blank'>Recipes for building an open-domain chatbot</a></p><p style='text-align: center'><a href='https://parl.ai/projects/recipes/' target='_blank'>Original PARLAI Code</a></p>"
# Text-to-text generation pipeline
generator = pipeline("text2text-generation", model="merve/chatgpt-prompts-bart-long")
# Funktion, die die Textgenerierung durchführt
def generate_prompt(prompt):
output = generator(prompt, max_length=250)
return output[0]['generated_text']
# Erstellung der Gradio-Schnittstelle
interface = gr.Interface(
fn=generate_prompt,
inputs=gr.inputs.Textbox(lines=2, placeholder="Enter your prompt here..."),
outputs="text",
title=title,
description=description,
article=article,
layout="vertical", # Setzt die Anordnung der Elemente auf vertikal
theme='ParityError/Interstellar'
)
# Starten der Schnittstelle
interface.launch()