Used better object detection test image. Fixed issue where err of Mak…

…eHFAPIRequestWithMedia err was not checked
Kardbord · Nov 23, 2021 · 9569024 · 9569024
1 parent dab3c02
commit 9569024
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/audio_classification.go b/audio_classification.go
@@ -12,6 +12,9 @@ type AudioClassificationResponse struct {
 
 func SendAudioClassificationRequest(model, audioFile string) ([]*AudioClassificationResponse, error) {
 	respBody, err := MakeHFAPIRequestWithMedia(model, audioFile)
+	if err != nil {
+		return nil, err
+	}
 
 	acresp := []*AudioClassificationResponse{}
 	err = json.Unmarshal(respBody, &acresp)

diff --git a/object_detection.go b/object_detection.go
@@ -26,6 +26,9 @@ type ObjectBox struct {
 
 func SendObjectDetectionRequest(model, imageFile string) ([]*ObjectDetectionResponse, error) {
 	respBody, err := MakeHFAPIRequestWithMedia(model, imageFile)
+	if err != nil {
+		return nil, err
+	}
 
 	resps := []*ObjectDetectionResponse{}
 	err = json.Unmarshal(respBody, &resps)

diff --git a/speech_recognition.go b/speech_recognition.go
@@ -15,6 +15,9 @@ type SpeechRecognitionResponse struct {
 // It reads the file and sends a request to the speech recognition endpoint.
 func SendSpeechRecognitionRequest(model, audioFile string) (*SpeechRecognitionResponse, error) {
 	respBody, err := MakeHFAPIRequestWithMedia(model, audioFile)
+	if err != nil {
+		return nil, err
+	}
 
 	arresp := SpeechRecognitionResponse{}
 	err = json.Unmarshal(respBody, &arresp)

diff --git a/test_files/test-image.jpeg b/test_files/test-image.jpeg