Spaces:

mateuscasare
/

deuxlex

Sleeping

mateuscasare commited on Jan 28

Commit

2a834a9

verified ·

1 Parent(s): c46768e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,9 +4,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForQuesti
 def carregar_modelos():
     # Caminhos no Hugging Face Hub
-    caminho_gpt = "mateuscasare/GPTNeo-Vade-Mecum-2023/GPT_fined_tuned"
-    caminho_bert = "mateuscasare/BERT-Vade-Mecum-2023/BERT_fine_tuned"
-    caminho_bert_juridico = "mateuscasare/modelo_juridico_pt/modelo_juridico_pt"
     dispositivo = "cuda" if torch.cuda.is_available() else "cpu"

 def carregar_modelos():
     # Caminhos no Hugging Face Hub
+    caminho_gpt = "mateuscasare/GPTNeo-Vade-Mecum-2023"
+    caminho_bert = "mateuscasare/BERT-Vade-Mecum-2023"
+    caminho_bert_juridico = "mateuscasare/modelo_juridico_pt"
     dispositivo = "cuda" if torch.cuda.is_available() else "cpu"