Update handler.py

handler.py

@@ -11,7 +11,7 @@ def install(package):
 
 class EndpointHandler:
     def __init__(self, path=""):
-        required_packages = ['timm', 'einops', 'flash-attn', 'Pillow','transformers==4.45.2']
+        required_packages = ['timm', 'einops', 'flash-attn', 'Pillow','transformers']
         for package in required_packages:
             try:
                 install(package)
@@ -22,7 +22,7 @@ class EndpointHandler:
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         print(f"Using device: {self.device}")
         
-        self.model_name = "arjunanand13/florence-enphaseall2-30e"
+        self.model_name = "microsoft/Florence-2-base"
         self.model = AutoModelForCausalLM.from_pretrained(
             self.model_name,
             trust_remote_code=True,