Update app.py

Moved model instantiation inside the function

app.py

@@ -2,14 +2,14 @@ from transformers import VisionEncoderDecoderModel
 from transformers import TrOCRProcessor, AutoTokenizer, ViTImageProcessor
 import gradio as gr
 
-model = VisionEncoderDecoderModel.from_pretrained("kavg/TrOCR-SIN")
-tokenizer = AutoTokenizer.from_pretrained('NLPC-UOM/SinBERT-large')
-feature_extractor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224")
-processor = TrOCRProcessor(image_processor=feature_extractor, tokenizer=tokenizer)
-
 def OCR(image):
     # Load the image
     img = Image.open(image)
+
+    model = VisionEncoderDecoderModel.from_pretrained("kavg/TrOCR-SIN")
+    tokenizer = AutoTokenizer.from_pretrained('NLPC-UOM/SinBERT-large')
+    feature_extractor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224")
+    processor = TrOCRProcessor(image_processor=feature_extractor, tokenizer=tokenizer)
     
     pixel_values = processor(img, return_tensors="pt").pixel_values
     generated_ids = model.generate(pixel_values)