Update app.py

app.py

@@ -3,14 +3,16 @@ import os
 from transformers import pipeline
 
 # Usa um modelo gratuito da Hugging Face (Llama 3 Instruct, ótimo para instruções)
+MODEL_NAME = "maritaca-ai/sabia-2-medium"  # Substitua por "meta-llama/Llama-3.1-8B-Instruct" se preferir
 gerador = pipeline(
-    "text-generation", 
-    model="mistralai/Mistral-7B-v0.1",
-    max_new_tokens=512,
-    token=os.getenv("HF_TOKEN"),  # opcional se estiver rodando via HF Spaces
-    temperature=0.6
+    "text-generation",
+    model=MODEL_NAME,
+    max_new_tokens=1024,
+    temperature=0.7,
+    top_p=0.9,
+    do_sample=True,
+    token=os.getenv("HF_TOKEN"),  # Necessário para modelos restritos
 )
-
 def responder_como_aldo(pergunta):
     prompt = f"""
 Você é o professor Dr. Aldo Henrique, especialista em C, Java, desenvolvimento web e inteligência artificial.