Update app.py

app.py

@@ -4,10 +4,11 @@ from PIL import Image
 import torch
 import io
 import os
+from typing import Union
 
 # Patch to remove flash-attn dependency
 from transformers.dynamic_module_utils import get_imports
-def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
+def fixed_get_imports(filename: Union[str, os.PathLike]) -> list[str]:
     """Work around for flash-attn imports."""
     if not str(filename).endswith("/modeling_florence2.py"):
         return get_imports(filename)
@@ -26,7 +27,6 @@ with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
 # Initialize FastAPI
 app = FastAPI()
 
-
 def process_image(image, task_token):
     inputs = processor(text=task_token, images=image, return_tensors="pt", padding=True).to(device)
     generated_ids = model.generate(