Update app.py

app.py

@@ -1,25 +1,22 @@
 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()
+def generate_response(text, command):
+    prompt = f"{command}: {text}"
+    result = generator(prompt, max_new_tokens=150, temperature=0.9, do_sample=True)
+    return result[0]["generated_text"].replace(prompt, "").strip()
 
-# Interface com Gradio
-demo = gr.Interface(
-    fn=gerar_resposta,
+interface = gr.Interface(
+    fn=generate_response,
     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"),
+    outputs=gr.Textbox(label="Resposta gerada"),
     title="GPT-2 App Sr. Nicolas",
-    description="Digite um texto e selecione o que deseja que a IA faça."
+    description="Digite um texto e selecione o comando desejado."
 )
 
-demo.launch()
+interface.launch()