Fix broken upload

modeling_cetacean_classifier.py → cetacean_classifier/modeling_cetacean_classifier.py

@@ -1,11 +1,12 @@
 import albumentations as A
 from transformers import PreTrainedModel
-from PIL import Image
+# from PIL import Image
 import numpy as np
 import torch
 import cv2
-from configuration_cetacean_classifier import CetaceanClassifierConfig
-from train import SphereClassifier
+
+from .train import SphereClassifier
+from .configuration_cetacean_classifier import CetaceanClassifierConfig
 
 
 WHALE_CLASSES = np.array(

train.py → cetacean_classifier/train.py

@@ -5,10 +5,10 @@ import timm
 import torch
 from pytorch_lightning import LightningDataModule, LightningModule, Trainer
 
-from config import Config, load_config
+from .config import Config, load_config
 # from .dataset import WhaleDataset, load_df
-from metric_learning import ArcFaceLossAdaptiveMargin, ArcMarginProductSubcenter, GeM
-from utils import WarmupCosineLambda, map_dict, topk_average_precision
+from .metric_learning import ArcFaceLossAdaptiveMargin, ArcMarginProductSubcenter, GeM
+from .utils import WarmupCosineLambda, map_dict, topk_average_precision
 
 
 class SphereClassifier(LightningModule):

push_to_hub.py

@@ -7,17 +7,17 @@ import json
 
 os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
 
-from configuration_cetacean_classifier import (
+from cetacean_classifier.configuration_cetacean_classifier import (
     CetaceanClassifierConfig,
 )
-from modeling_cetacean_classifier import (
+from cetacean_classifier.modeling_cetacean_classifier import (
     CetaceanClassifierModelForImageClassification,
 )
 
 CetaceanClassifierConfig.register_for_auto_class()
 CetaceanClassifierModelForImageClassification.register_for_auto_class("AutoModelForImageClassification")
 
-with open("orignal_model_config.json", "r") as file:
+with open("original_model_config.json", "r") as file:
     config= json.load(file)
 
 cetacean_config = CetaceanClassifierConfig(**config)

test_model_before_push.py

@@ -8,10 +8,10 @@ import cv2
 
 os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
 
-from configuration_cetacean_classifier import (
+from cetacean_classifier.configuration_cetacean_classifier import (
     CetaceanClassifierConfig,
 )
-from modeling_cetacean_classifier import (
+from cetacean_classifier.modeling_cetacean_classifier import (
     CetaceanClassifierModelForImageClassification,
 )