feat: Update re-layout logic for model serving

cmd/ormb-storage-initializer/cmd/pull-and-export.go

@@ -17,18 +17,15 @@ package cmd
 
 import (
 	"fmt"
-	"io/ioutil"
 	"os"
 	"path"
 	"path/filepath"
 	"strings"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
-	"gopkg.in/yaml.v2"
 
 	"github.com/kleveross/ormb/pkg/consts"
-	"github.com/kleveross/ormb/pkg/model"
 	"github.com/kleveross/ormb/pkg/oras"
 	"github.com/kleveross/ormb/pkg/ormb"
 )
@@ -110,42 +107,11 @@ func relayoutModel(modelDir string) error {
 	if err != nil {
 		return err
 	}
-	destinationDir, err := filepath.Abs(modelDir)
-	if err != nil {
-		return err
-	}
 
-	// For TensorFlow serving, MUST move /mnt/models/model to /mnt/models/1 (dstDir).
-	ormbfileBytes, err := ioutil.ReadFile(path.Join(modelDir, "ormbfile.yaml"))
-	if err != nil {
+	if err := os.Rename(originalDir, path.Join(modelDir, "1")); err != nil {
 		return err
 	}
-	var metadata model.Metadata
-	err = yaml.Unmarshal(ormbfileBytes, &metadata)
-	if err != nil {
-		return err
-	}
-	if metadata.Format == string(model.FormatSavedModel) {
-		if err := os.Rename(originalDir, path.Join(modelDir, "1")); err != nil {
-			return err
-		}
-
-		return nil
-	}
 
-	// For other format model.
-	files, err := ioutil.ReadDir(originalDir)
-	if err != nil {
-		return err
-	}
-	for _, f := range files {
-		oldPath := filepath.Join(originalDir, f.Name())
-		newPath := filepath.Join(destinationDir, f.Name())
-		fmt.Printf("Moving %s to %s\n", oldPath, newPath)
-		if err := os.Rename(oldPath, newPath); err != nil {
-			return err
-		}
-	}
 	return nil
 }