bentobytes commited on
Commit
4d05c0d
·
1 Parent(s): cdfa5ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import re
2
  import gradio as gr
 
3
 
4
  import torch
5
  from transformers import DonutProcessor, VisionEncoderDecoderModel
@@ -51,4 +52,10 @@ demo = gr.Interface(
51
  examples=[["example.png"], ["example_1.png"],["example_2.png"], ["example_3.png"],["example_4.png"]],
52
  cache_examples=False)
53
 
54
- demo.launch()
 
 
 
 
 
 
 
1
  import re
2
  import gradio as gr
3
+ import os
4
 
5
  import torch
6
  from transformers import DonutProcessor, VisionEncoderDecoderModel
 
52
  examples=[["example.png"], ["example_1.png"],["example_2.png"], ["example_3.png"],["example_4.png"]],
53
  cache_examples=False)
54
 
55
+ credentials_json = os.environ.get("CREDENTIALS")
56
+
57
+ if credentials_json is None:
58
+ print("Error: Please set the CREDENTIALS")
59
+ else:
60
+ credentials = json.loads(credentials_json)
61
+ demo.launch(auth=credentials)