Upload model

Model.py

@@ -103,7 +103,7 @@ class InceptionV3ModelForImageClassification(PreTrainedModel):
 			image = Image.open(path).convert('RGB')
 		return image
 
-	def predict(self, path, device="cuda"):
+	def predict(self, path, device="cuda", print_tensor=True):
 		image = self.open_image(path)
 		image = self.transform(image)
 		image = image.unsqueeze(0)
@@ -116,6 +116,7 @@ class InceptionV3ModelForImageClassification(PreTrainedModel):
 			self.cpu()
 		with torch.no_grad():
 			out, aux = self(image)
-			print(out)
+			if print_tensor:
+				print(out)
 			_, predicted = torch.max(out.data, 1)
 		return self.config.classes[predicted.item()]