MinhaPrompt / app.py
Guilbert12's picture
Update app.py
c39baa4 verified
Raw History Blame Contribute Delete
669 Bytes
import gradio as gr
from transformers import pipeline
# Cria o gerador de texto com o modelo GPT-2
gerador = pipeline("text-generation", model="gpt2")
def gerar_prompt(entrada):
prompt = f"Crie um prompt detalhado de vídeo ultra-realista baseado nisso: {entrada}\n"
resultado = gerador(prompt, max_length=250, num_return_sequences=1)
return resultado[0]['generated_text']
# Interface gráfica do Gradio
iface = gr.Interface(
fn=gerar_prompt,
inputs="text",
outputs="text",
title="Minha IA de Prompts para Vídeos Realistas",
description="Digite sua ideia de vídeo e a IA cria um prompt detalhado pronto para usar."
)
iface.launch()