tidy up code

yansigit · yansigit · commit 57158028e68f · 2025-03-02T22:08:02.000-07:00
diff --git a/main.go b/main.go
@@ -74,7 +74,7 @@ func main() {
 		return
 	}
 
-	fmt.Printf("Model downloaded successfully to %s\n", outputPath)
+	fmt.Printf("Model downloaded successfully to %s\n", filepath.Dir(outputPath))
 }
 
 func loadConfig(filename string) (*Config, error) {
@@ -108,8 +108,7 @@ func downloadFile(outputPath string, url string) error {
 		return fmt.Errorf("HTTP error for %s: %v", url, resp.StatusCode)
 	}
 
-	// Check Content-Disposition header for filename
-	header := resp.Header.Get("Content-Disposition")
+	header := resp.Header.Get("content-disposition")
 	if header != "" {
 		parts := strings.Split(header, "filename=")
 		if len(parts) > 1 {

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ func main() {`
`74`	`74`	`return`
`75`	`75`	`}`
`76`	`76`
`77`		`- fmt.Printf("Model downloaded successfully to %s\n", outputPath)`
	`77`	`+ fmt.Printf("Model downloaded successfully to %s\n", filepath.Dir(outputPath))`
`78`	`78`	`}`
`79`	`79`
`80`	`80`	`func loadConfig(filename string) (*Config, error) {`
`@@ -108,8 +108,7 @@ func downloadFile(outputPath string, url string) error {`
`108`	`108`	`return fmt.Errorf("HTTP error for %s: %v", url, resp.StatusCode)`
`109`	`109`	`}`
`110`	`110`
`111`		`- // Check Content-Disposition header for filename`
`112`		`- header := resp.Header.Get("Content-Disposition")`
	`111`	`+ header := resp.Header.Get("content-disposition")`
`113`	`112`	`if header != "" {`
`114`	`113`	`parts := strings.Split(header, "filename=")`
`115`	`114`	`if len(parts) > 1 {`