FredZhang7
/

google-safesearch-mini-tfjs

Image Classification

Model card Files Files and versions

FredZhang7 commited on Dec 27, 2022

Commit

7739462

·

1 Parent(s): b0e1880

Upload model

Files changed (1) hide show

Model.py +4 -1

Model.py CHANGED Viewed

@@ -10,7 +10,10 @@ class InceptionV3ModelForImageClassification(PreTrainedModel):
 		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:
 			raise ValueError(f"Model {self.config.model_name} not found.")

 		if self.config.model_name == "google-safesearch-mini":
 			if not os.path.exists(model_path):
+				import requests
+				url = "https://huggingface.co/FredZhang7/google-safesearch-mini/resolve/main/pytorch_model.bin"
+				r = requests.get(url, allow_redirects=True)
+				open(model_path, 'wb').write(r.content)
 			self.model = torch.jit.load(model_path)
 		else:
 			raise ValueError(f"Model {self.config.model_name} not found.")