Update handler.py

handler.py

@@ -9,8 +9,8 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.model = Swinv2Model.from_pretrained("microsoft/moooji/swinv2-large-patch4-window12to24-192to384-22kto1k-ft").to(device)
-        self.processor = AutoImageProcessor.from_pretrained("microsoft/moooji/swinv2-large-patch4-window12to24-192to384-22kto1k-ft")
+        self.model = Swinv2Model.from_pretrained("microsoft/swinv2-large-patch4-window12to24-192to384-22kto1k-ft").to(device)
+        self.processor = AutoImageProcessor.from_pretrained("microsoft/swinv2-large-patch4-window12to24-192to384-22kto1k-ft")
 
     def __call__(self, data: Any) -> List[float]:
         inputs = data.pop("inputs", data)