added PNG and latex code

app.py

@@ -1,8 +1,33 @@
 from fastapi import FastAPI
+from transformers import TrOCRProcessor, VisionEncoderDecoderModel
+from PIL import Image
 
 app = FastAPI()
 
 @app.get("/")
 def greet_json():
-    return {"Hello": "World!"}
+    # Load model and processor from Hugging Face
+    processor = TrOCRProcessor.from_pretrained('tjoab/latex_finetuned')
+    model = VisionEncoderDecoderModel.from_pretrained('tjoab/latex_finetuned')
 
+
+    # Load all images as a batch
+    images = [open_PIL_image(path) for path in paths]
+
+    # Preprocess the images 
+    preproc_image = processor.image_processor(images=images, return_tensors="pt").pixel_values
+
+    # Generate and decode the tokens
+    # NOTE: max_length default value is very small, which often results in truncated inference if not set 
+    pred_ids = model.generate(preproc_image, max_length=128)
+    latex_preds = processor.batch_decode(pred_ids, skip_special_tokens=True)
+    return {"message": "Success", "latex_preds": latex_preds}
+
+
+
+# Helper funtion (path to either JPEG or PNG)
+def open_PIL_image(image_path: str) -> Image.Image:
+  image = Image.open(image_path)
+  if image_path.split('.')[-1].lower() == 'png':
+      image = Image.composite(image, PIL.Image.new('RGB', image.size, 'white'), image)
+  return image