Update src/brain.py

src/brain.py

@@ -1,7 +1,7 @@
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-tokenizer = GPT2Tokenizer.from_pretrained("distilgpt2")
-model = GPT2LMHeadModel.from_pretrained("distilgpt2")
+tokenizer = GPT2Tokenizer.from_pretrained('juridics/bertimbaulaw-base-portuguese-sts-scale')
+model = GPT2LMHeadModel.from_pretrained('juridics/bertimbaulaw-base-portuguese-sts-scale')
 
 def generate_answers(query):
     input_ids = tokenizer.encode(query, return_tensors='pt')