Update app.py

app.py

@@ -29,13 +29,17 @@ def answer_question_with_gpt(question):
         input_ids,
         attention_mask=attention_mask,
         max_length=len(input_ids[0]) + 100,  # Aumentar o limite de geração
-        temperature=0.7,  # Ajustar a criatividade
+        temperature=0.75,  # Ajustar a criatividade
         top_p=0.9,  # Usar nucleus sampling
         no_repeat_ngram_size=2  # Evitar repetições desnecessárias
     )
     
     generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
-    return generated_text
+    # Processando para extrair apenas a resposta após "Resposta:"
+    response_part = generated_text.split("Resposta:")[1] if "Resposta:" in generated_text else "Resposta não encontrada."
+    # Limpeza adicional para remover qualquer texto indesejado após a resposta
+    final_response = response_part.split(".")[0] + "."  # Isso assume que a resposta termina na primeira sentença.
+    return final_response
 
 # Função para adicionar feedback
 def add_feedback(nome, feedback):