diff --git a/pkg/predict/input.go b/pkg/predict/input.go
index 113111eba8..e2fba26fbc 100644
--- a/pkg/predict/input.go
+++ b/pkg/predict/input.go
@@ -2,13 +2,14 @@ package predict
 
 import (
 	"fmt"
-	"mime"
 	"os"
 	"path/filepath"
 	"strings"
 
 	"github.com/mitchellh/go-homedir"
 	"github.com/vincent-petithory/dataurl"
+
+	"github.com/replicate/cog/pkg/util/mime"
 )
 
 type Input struct {