apruvd commited on
Commit
8426bb8
·
1 Parent(s): 588a558

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,3 +1,5 @@
1
  import gradio as gr
2
- #
 
 
3
  gr.load("models/apruvd/tk",api_key = "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").launch()
 
1
  import gradio as gr
2
+ import os
3
+ auth_token = os.environ.get("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") or True
4
+ AutoModelForMaskedLM.from_pretrained("apruvd/tk", use_auth_token=auth_token)
5
  gr.load("models/apruvd/tk",api_key = "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").launch()