JeffersonCorreiax commited on
Commit
f3f1d25
·
verified ·
1 Parent(s): 4bd23ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -9
app.py CHANGED
@@ -20,23 +20,21 @@ app.add_middleware(
20
  )
21
 
22
 
23
- #Carrega o token do Hugging Face a partir do .env
24
  load_dotenv()
25
  hf_token = os.getenv("HF_TOKEN")
26
 
27
  if not hf_token:
28
  raise ValueError("Token do Hugging Face não encontrado. Adicione HF_TOKEN no arquivo .env")
29
 
30
- #Nome do modelo privado
31
- model_name = "NullisTerminis/oncology_t5"
32
-
33
- #Autenticação no Hugging Face
34
-
35
  login(token=hf_token)
36
 
37
- #Carregar o modelo e o tokenizer
38
- tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
39
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=hf_token)
 
 
40
 
41
  #Teste simples
42
  #input_text = input(str("Texto:"))
 
20
  )
21
 
22
 
23
+ # Carrega o token do Hugging Face a partir do .env
24
  load_dotenv()
25
  hf_token = os.getenv("HF_TOKEN")
26
 
27
  if not hf_token:
28
  raise ValueError("Token do Hugging Face não encontrado. Adicione HF_TOKEN no arquivo .env")
29
 
30
+ # Autentica (opcional se já tiver setado a env var)
 
 
 
 
31
  login(token=hf_token)
32
 
33
+ model_name = "NullisTerminis/oncology_t5"
34
+
35
+ # Carrega o modelo usando o token atualizado
36
+ tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
37
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name, token=hf_token)
38
 
39
  #Teste simples
40
  #input_text = input(str("Texto:"))