Upload model

Model.py

@@ -2,7 +2,6 @@ from transformers import PreTrainedModel
 import torch
 import os
 
-
 class InceptionV3ModelForImageClassification(PreTrainedModel):
 	def __init__(self, config):
 		super().__init__(config)
@@ -10,12 +9,11 @@ class InceptionV3ModelForImageClassification(PreTrainedModel):
 		model_path = f"{self.config.model_name}.bin".replace("/","_")
 
 		if self.config.model_name == "google-safesearch-mini":
+			if not os.path.exists(model_path):
+				model_path = f"./InceptionV3/{model_path}"
 			self.model = torch.jit.load(model_path)
 		else:
-			if not os.path.exists(model_path):
-				from urllib.request import urlretrieve
-				urlretrieve(f"https://huggingface.co/{self.config.model_name}/resolve/main/pytorch_model.bin", model_path)
-			self.model = torch.jit.load(model_path) if self.config.use_jit else torch.load(model_path)
+			raise ValueError(f"Model {self.config.model_name} not found.")
 
 	def forward(self, input_ids):
 		return self.model(input_ids), None if self.config.model_name == "inception_v3" else self.model(input_ids)