TEST_8

app.py

@@ -4,16 +4,12 @@ from sentence_transformers import SentenceTransformer, util
 
 class ModelComparator:
     def __init__(self):
-        # Modelo de QA (mais rápido e leve)
         self.qa_pipeline = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
-        # Modelo de geração de texto simples
-        self.text_gen_pipeline = pipeline("text-generation", model="gpt2", max_new_tokens=50)
-        # Modelo para embeddings e similaridade
+        self.text_gen_pipeline = pipeline("text-generation", model="gpt2", max_new_tokens=20)  # menor geração
         self.sim_model = SentenceTransformer("all-MiniLM-L6-v2")
 
     def get_qa_answer(self, question, context=None):
-        # Se não passar contexto, responde "não sei"
-        if context is None:
+        if not context:
             return "No context provided for QA model."
         try:
             result = self.qa_pipeline(question=question, context=context)
@@ -24,7 +20,6 @@ class ModelComparator:
     def get_text_gen_answer(self, prompt):
         try:
             generated = self.text_gen_pipeline(prompt)[0]['generated_text']
-            # O GPT2 gera o texto incluindo o prompt, vamos remover o prompt para deixar só resposta
             answer = generated[len(prompt):].strip()
             return answer if answer else generated.strip()
         except Exception as e:
@@ -45,14 +40,12 @@ class ModelComparator:
                 f"Model GPT-2 generated answer:\n{gen_answer}\n\n"
                 f"Semantic similarity score: {similarity}")
 
-# Interface Gradio
-
 model_comparator = ModelComparator()
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Comparador de respostas entre dois modelos locais (CPU)")
+    gr.Markdown("## Comparador rápido para Hugging Face Spaces")
     question_input = gr.Textbox(label="Pergunta")
-    context_input = gr.Textbox(label="Contexto para o modelo de QA (opcional)", lines=5)
+    context_input = gr.Textbox(label="Contexto para o modelo de QA (opcional)", lines=3)
     output = gr.Textbox(label="Respostas e Similaridade", lines=15)
     btn = gr.Button("Comparar")