Node_Server

Sleeping

freealise commited on Dec 24, 2024

Commit

31d6614

verified ·

1 Parent(s): d1ad894

Update index.js

Files changed (1) hide show

index.js CHANGED Viewed

@@ -9,7 +9,7 @@ class MyClassificationPipeline {
     if (this.instance === null) {
       // NOTE: Uncomment this to change the cache directory
-      // env.cacheDir = './.cache';
       this.instance = pipeline(this.task, this.model, { progress_callback });
     }

     if (this.instance === null) {
       // NOTE: Uncomment this to change the cache directory
+      env.cacheDir = './.cache';
       this.instance = pipeline(this.task, this.model, { progress_callback });
     }