Darwin Danish commited on
Commit
bd1ee30
·
verified ·
1 Parent(s): 92f80da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,8 +1,10 @@
1
  import gradio as gr
2
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
  import json
 
 
4
 
5
- # Load the tokenizer and model
6
  tokenizer = AutoTokenizer.from_pretrained("Darwin29/lumina-translate")
7
  model = AutoModelForSeq2SeqLM.from_pretrained("Darwin29/lumina-translate")
8
 
 
1
  import gradio as gr
2
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
3
  import json
4
+ import os
5
+ os.environ["HF_TOKEN"] = os.getenv('MODEL_REPO_ID')
6
 
7
+ # Load the tokenizer and model from Hugging Face (authentication handled via token)
8
  tokenizer = AutoTokenizer.from_pretrained("Darwin29/lumina-translate")
9
  model = AutoModelForSeq2SeqLM.from_pretrained("Darwin29/lumina-translate")
10