Update app.py

app.py

@@ -1,9 +1,10 @@
 import pandas as pd
 import gradio as gr
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import AutoTokenizer, AutoModel
+import torch
 
-tokenizer = GPT2Tokenizer.from_pretrained("distilgpt2")
-model = GPT2LMHeadModel.from_pretrained("distilgpt2")
+tokenizer = AutoTokenizer.from_pretrained('juridics/bertimbaulaw-base-portuguese-sts-scale')
+model = AutoModel.from_pretrained('juridics/bertimbaulaw-base-portuguese-sts-scale')
 
 # Dados iniciais
 data = {