Update app.py

app.py

@@ -1,35 +1,34 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, set_seed
 
-# Carregando os dois modelos de geração de texto
-modelo_1 = pipeline("text-generation", model="gpt2")
-modelo_2 = pipeline("text-generation", model="distilgpt2")
+# Garante um resultado mais consistente
+set_seed(42)
 
-# Função avaliadora simples: escolhe a resposta mais longa
-def avaliar_respostas(prompt, resposta1, resposta2):
+# Modelos em português
+modelo_1 = pipeline("text-generation", model="pierreguillou/gpt2-small-portuguese")
+modelo_2 = pipeline("text-generation", model="neuralmind/gpt2-medium-portuguese")
+
+# Função que avalia as respostas (escolhe a mais longa, por simplicidade)
+def avaliar_respostas(resposta1, resposta2):
     if len(resposta1) > len(resposta2):
         return f"Resposta escolhida do Modelo 1:\n\n{resposta1}"
     else:
         return f"Resposta escolhida do Modelo 2:\n\n{resposta2}"
 
-# Função principal do chatbot em cascata
-def chatbot_em_cascata(prompt):
-    # Gerar respostas com os dois modelos
-    resposta1 = modelo_1(prompt, max_length=50, do_sample=True)[0]['generated_text']
-    resposta2 = modelo_2(prompt, max_length=50, do_sample=True)[0]['generated_text']
-
-    # Avaliar e escolher a melhor resposta
-    melhor_resposta = avaliar_respostas(prompt, resposta1, resposta2)
-    return melhor_resposta
+# Função principal
+def responder(prompt):
+    resp1 = modelo_1(prompt, max_length=60, num_return_sequences=1)[0]['generated_text']
+    resp2 = modelo_2(prompt, max_length=60, num_return_sequences=1)[0]['generated_text']
+    melhor = avaliar_respostas(resp1, resp2)
+    return melhor
 
-# Interface com Gradio
-interface = gr.Interface(
-    fn=chatbot_em_cascata,
-    inputs=gr.Textbox(lines=2, placeholder="Digite sua pergunta aqui..."),
-    outputs="text",
-    title="Chatbot em Cascata - FMU Engenharia",
-    description="Dois modelos geram respostas, um avaliador escolhe a melhor."
-)
+# Interface Gradio
+gr.Interface(
+    fn=responder,
+    inputs=gr.Textbox(label="Digite sua pergunta"),
+    outputs=gr.Textbox(label="Resposta escolhida"),
+    title="Chatbot em Português - FMU",
+    description="Dois modelos em português geram respostas; o melhor é exibido."
+).launch()
 
-interface.launch()