Hugging Face's logo

Spaces:
mateuscasare
/
deuxlex
Sleeping

App Files Community
mateuscasare commited on
Commit
c46768e
·
verified ·
1 Parent(s): 76979e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForQuesti
4
 
5
  def carregar_modelos():
6
  # Caminhos no Hugging Face Hub
7
- caminho_gpt = "mateuscasare/GPTNeo-Vade-Mecum-2023/GPTNeo-Vade-Mecum-2023"
8
- caminho_bert = "mateuscasare/BERT-Vade-Mecum-2023/BERT-Vade-Mecum-2023"
9
  caminho_bert_juridico = "mateuscasare/modelo_juridico_pt/modelo_juridico_pt"
10
 
11
  dispositivo = "cuda" if torch.cuda.is_available() else "cpu"
 
4
 
5
  def carregar_modelos():
6
  # Caminhos no Hugging Face Hub
7
+ caminho_gpt = "mateuscasare/GPTNeo-Vade-Mecum-2023/GPT_fined_tuned"
8
+ caminho_bert = "mateuscasare/BERT-Vade-Mecum-2023/BERT_fine_tuned"
9
  caminho_bert_juridico = "mateuscasare/modelo_juridico_pt/modelo_juridico_pt"
10
 
11
  dispositivo = "cuda" if torch.cuda.is_available() else "cpu"