Update app.py

app.py

@@ -6,9 +6,16 @@ from torchvision.transforms import functional as F
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 from transformers import AutoModel
 
+from transformers import AutoModel
+
+model_name = "trocrnew.pth"
 access_token = os.environ.get("HF_TOKEN")
+model = AutoModel.from_pretrained(model_name, token=access_token)
+
+# access_token = os.environ.get("HF_TOKEN")
+
+# model = AutoModel.from_pretrained("trocrnew.pth", token=access_token)
 
-model = AutoModel.from_pretrained("trocrnew.pth", token=access_token)
 app = Flask(__name__)
 
 # Load the trained model and processor