app.py

app.py

@@ -0,0 +1,25 @@
+from transformers import pipeline
+import gradio as gr
+
+# Cria pipeline de geração de texto com GPT-2
+generator = pipeline("text-generation", model="gpt2")
+
+# Função principal da IA
+def gerar_resposta(texto, comando):
+    prompt = f"{comando}: {texto}"
+    resultado = generator(prompt, max_new_tokens=150, temperature=0.9, do_sample=True)
+    return resultado[0]["generated_text"].replace(prompt, "").strip()
+
+# Interface com Gradio
+demo = gr.Interface(
+    fn=gerar_resposta,
+    inputs=[
+        gr.Textbox(lines=5, label="Texto de entrada"),
+        gr.Dropdown(choices=["Expandir", "Resumir", "Simplificar", "Continuar"], label="Comando")
+    ],
+    outputs=gr.Textbox(label="Texto gerado pela IA"),
+    title="GPT-2 App Sr. Nicolas",
+    description="Digite um texto e selecione o que deseja que a IA faça."
+)
+
+demo.launch()