sk2003 commited on
Commit
151d09d
·
1 Parent(s): 9a12092

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ from transformers import TrOCRProcessor, VisionEncoderDecoderModel # importing t
11
  import gradio as gr
12
 
13
  processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
14
- model = VisionEncoderDecoderModel.from_pretrained("sk2003/hist-trocr")
 
15
  def process_image(image):
16
  # prepare image
17
  pixel_values = processor(image, return_tensors="pt").pixel_values
 
11
  import gradio as gr
12
 
13
  processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
14
+ auth_token = os.environ.get("TOKEN_FROM_SECRET") or True
15
+ model = VisionEncoderDecoderModel.from_pretrained("sk2003/hist-trocr", use_auth_token=auth_token)
16
  def process_image(image):
17
  # prepare image
18
  pixel_values = processor(image, return_tensors="pt").pixel_values