robzchhangte commited on
Commit
f331fc9
1 Parent(s): eeec3d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,12 +1,13 @@
 
1
  import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
- from huggingface_hub import login
4
 
 
5
 
6
  # Load the translation model and tokenizer from Hugging Face
7
  model_name = "robzchhangte/enmz-helsinki-case"
8
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
9
- tokenizer = AutoTokenizer.from_pretrained(model_name)
10
 
11
  # Translation function with max_length=512
12
  def translate(text):
 
1
+ import os
2
  import gradio as gr
3
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
4
 
5
+ token = os.getenv('MODEL_REPO_ID')).
6
 
7
  # Load the translation model and tokenizer from Hugging Face
8
  model_name = "robzchhangte/enmz-helsinki-case"
9
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=token)
10
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=token)
11
 
12
  # Translation function with max_length=512
13
  def translate(text):