DigiP-AI commited on
Commit
4d07dfc
·
verified ·
1 Parent(s): fc19c66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -45,7 +45,8 @@ theme = gr.themes.Soft(
45
  button_cancel_background_fill_hover='*primary_400'
46
  )
47
  #----------End of theme----------
48
-
 
49
 
50
  API_TOKEN = os.getenv("HF_READ_TOKEN")
51
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
45
  button_cancel_background_fill_hover='*primary_400'
46
  )
47
  #----------End of theme----------
48
+ auth_token = os.environ.get("TOKEN_FROM_SECRET") or True
49
+ AutoModelForMaskedLM.from_pretrained("DigiP-AI/film-noir", use_auth_token=auth_token)
50
 
51
  API_TOKEN = os.getenv("HF_READ_TOKEN")
52
  headers = {"Authorization": f"Bearer {API_TOKEN}"}