fixes

cmd/main_test.go

@@ -3,6 +3,7 @@ package main
 import (
 	"context"
 	"fmt"
+	"log"
 	"os"
 	"path"
 	"testing"
@@ -21,8 +22,18 @@ var textClassificationData []byte
 var tokenClassificationData []byte
 
 func TestMain(m *testing.M) {
+	entries, err := os.ReadDir("../models")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	for _, e := range entries {
+		fmt.Println(e.Name())
+	}
+
 	// model setup
 	if ok, err := util.FileSystem.Exists(context.Background(), "../models"); err == nil {
+		fmt.Println("HERE")
 		if !ok {
 			session, err := hugot.NewSession()
 			if err != nil {
@@ -81,6 +92,7 @@ func TestTextClassificationCli(t *testing.T) {
 
 	args := append(baseArgs, "run", fmt.Sprintf("--input=%s", testDataDir), fmt.Sprintf("--model=%s", testModel), "--type=textClassification")
 	if err := app.Run(args); err != nil {
+		fmt.Println("ERROR HERE")
 		check(t, err)
 	}
 }