Upload model

model.py

@@ -9,17 +9,17 @@ class SafeSearchConfig(PretrainedConfig):
 				 model_name: str = "safesearch_mini_v2",
 				 input_channels: int = 3,
 				 num_classes: int = 3,
-				 input_size: list[int] = [3, 299, 299],
-				 pool_size: list[int] = [8, 8],
+				 input_size: list = [3, 299, 299],
+				 pool_size: list = [8, 8],
 				 crop_pct: float = 0.875,
 				 interpolation: str = "bicubic",
-				 mean: list[float] = [0.5, 0.5, 0.5],
-				 std: list[float] = [0.5, 0.5, 0.5],
+				 mean: list = [0.5, 0.5, 0.5],
+				 std: list = [0.5, 0.5, 0.5],
 				 first_conv: str = "conv2d",
 				 classifier: str = "default",
 				 has_aux: bool = False,
 				 label_offset: int = 0,
-				 classes: list[str] = classes,
+				 classes: object = classes,
 				 output_channels: int = 1536,
 				 device: str = "cpu",
 				 **kwargs):