Update pipeline.py

pipeline.py

@@ -12,8 +12,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 class PreTrainedPipeline():
     def __init__(self, path=""):
         self.model = ViTModel.from_pretrained(
-            pretrained_model_name_or_path=os.path.join(
-                path, 'pytorch_model.bin'),
+            pretrained_model_name_or_path=path,
             config=os.path.join(path, 'config.json')
         )
         self.model.eval()