Update app.py

app.py

@@ -1,4 +1,4 @@
-from transformers import AutoTokenizer, AutoModelForImageClassification
+from transformers import AutoTokenizer, AutoModel
 
 def load_model_and_tokenizer():
     try:
@@ -10,7 +10,7 @@ def load_model_and_tokenizer():
         )
 
         # Load the model with the specific revision
-        model = AutoModelForImageClassification.from_pretrained(
+        model = AutoModel.from_pretrained(
             'stepfun-ai/GOT-OCR2_0',
             revision='cf6b7386bc89a54f09785612ba74cb12de6fa17c',  # Pin the specific revision
             trust_remote_code=True
@@ -28,4 +28,4 @@ if __name__ == "__main__":
     if model and tokenizer:
         print("Model and tokenizer loaded successfully!")
     else:
-        print("Failed to load model and tokenizer.")
+        print("Failed to load model and tokenizer.")