Update app.py

app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, UploadFile, File
-from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers import AutoProcessor, AutoModelForVision2Seq,AutoModel
 from PIL import Image
 import torch
 import io
@@ -16,15 +16,25 @@ print("Loading GLM-OCR model...")
 #     torch_dtype=torch.float32
 # )
 
+# processor = AutoProcessor.from_pretrained(
+#     MODEL_ID,
+#     trust_remote_code=True
+# )
+
+# model = AutoModelForVision2Seq.from_pretrained(
+#     MODEL_ID,
+#     trust_remote_code=True,
+#     torch_dtype=torch.float32
+# )
+
 processor = AutoProcessor.from_pretrained(
     MODEL_ID,
     trust_remote_code=True
 )
 
-model = AutoModelForVision2Seq.from_pretrained(
+model = AutoModel.from_pretrained(
     MODEL_ID,
-    trust_remote_code=True,
-    torch_dtype=torch.float32
+    trust_remote_code=True
 )
 
 @app.get("/")