Nick-2x commited on
Commit
757def3
·
verified ·
1 Parent(s): 08f0ec2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -10,9 +10,20 @@ MODEL_ID = "zai-org/GLM-OCR"
10
 
11
  print("Loading GLM-OCR model...")
12
 
13
- processor = AutoProcessor.from_pretrained(MODEL_ID)
 
 
 
 
 
 
 
 
 
 
14
  model = AutoModelForVision2Seq.from_pretrained(
15
  MODEL_ID,
 
16
  torch_dtype=torch.float32
17
  )
18
 
 
10
 
11
  print("Loading GLM-OCR model...")
12
 
13
+ # processor = AutoProcessor.from_pretrained(MODEL_ID)
14
+ # model = AutoModelForVision2Seq.from_pretrained(
15
+ # MODEL_ID,
16
+ # torch_dtype=torch.float32
17
+ # )
18
+
19
+ processor = AutoProcessor.from_pretrained(
20
+ MODEL_ID,
21
+ trust_remote_code=True
22
+ )
23
+
24
  model = AutoModelForVision2Seq.from_pretrained(
25
  MODEL_ID,
26
+ trust_remote_code=True,
27
  torch_dtype=torch.float32
28
  )
29