Upload folder using huggingface_hub

script.py

@@ -8,7 +8,7 @@ import torchvision.transforms as T
 from PIL import Image
 import torch
 from transformers import AutoImageProcessor
-from submission.create_model import HieraForImageClassification
+from create_model import HieraForImageClassification
 #%%
 # %%
 
@@ -28,8 +28,8 @@ class PytorchWorker:
             self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
             print(f"Using devide: {self.device}")
 
-            image_processor = AutoImageProcessor.from_pretrained("./submission/hiera_model/")
-            model = HieraForImageClassification.from_pretrained("./submission/hiera_model/", num_labels =1784 ).to(self.device).eval()
+            image_processor = AutoImageProcessor.from_pretrained("./hiera_model/")
+            model = HieraForImageClassification.from_pretrained("./hiera_model/", num_labels =1784 ).to(self.device).eval()
 
             return model, image_processor