Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import shutil
 import os
 from zipfile import ZipFile
-from transformers import RobertaTokenizer, RobertaForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
 # Diretório temporário para os uploads
@@ -12,8 +12,8 @@ UPLOAD_DIR = "uploaded_files"
 java_files_content = {}
 
 # Carregar o modelo e o tokenizador do CodeBERT localmente
-tokenizer = RobertaTokenizer.from_pretrained("microsoft/codebert-base")
-model = RobertaForSequenceClassification.from_pretrained("microsoft/codebert-base")
+tokenizer = AutoTokenizer.from_pretrained("microsoft/codebert-base")
+model = AutoModelForSequenceClassification.from_pretrained("microsoft/codebert-base")
 
 # Função para processar a pasta enviada
 def process_uploaded_folder(zip_path):