toiar commited on
Commit
51594f5
·
verified ·
1 Parent(s): c37cb4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -2,11 +2,8 @@ import gradio as gr
2
  import os
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
- from huggingface_hub import login
6
 
7
  HF_TOKEN = os.environ.get("HF_TOKEN")
8
- if HF_TOKEN:
9
- login(token=HF_TOKEN)
10
 
11
  MODEL_ID = "toiar/nllb-finetuned-english-pnar"
12
 
@@ -16,6 +13,7 @@ tokenizer = AutoTokenizer.from_pretrained(
16
  )
17
  tokenizer.src_lang = "eng_Latn"
18
 
 
19
  model = AutoModelForSeq2SeqLM.from_pretrained(
20
  MODEL_ID,
21
  token=HF_TOKEN,
 
2
  import os
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
5
 
6
  HF_TOKEN = os.environ.get("HF_TOKEN")
 
 
7
 
8
  MODEL_ID = "toiar/nllb-finetuned-english-pnar"
9
 
 
13
  )
14
  tokenizer.src_lang = "eng_Latn"
15
 
16
+ # Load model
17
  model = AutoModelForSeq2SeqLM.from_pretrained(
18
  MODEL_ID,
19
  token=HF_TOKEN,