freemt commited on
Commit
d66b767
1 Parent(s): bdf7057

Update model_s.py

Browse files
Files changed (5) hide show
  1. .gitattributes +1 -0
  2. .gitignore +1 -0
  3. app.exe +0 -0
  4. chart_df.png +0 -0
  5. model_s.py +17 -0
.gitattributes CHANGED
@@ -25,3 +25,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ model_s.lzma filter=lfs diff=lfs merge=lfs -text
.gitignore CHANGED
@@ -4,3 +4,4 @@ app.build
4
  app.dist
5
  build
6
  app.exe
 
4
  app.dist
5
  build
6
  app.exe
7
+ model_s.lzma
app.exe DELETED
Binary file (702 kB)
chart_df.png DELETED
Binary file (13.3 kB)
model_s.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Load model_s."""
2
+ # pylint: disable=invalid-name
3
+
4
+ from pathlib import Path
5
+ import joblib
6
+
7
+ # from sentence_transformers import SentenceTransformer
8
+ from alive_progress import alive_bar
9
+
10
+ file_loc = "radiobee/model_s.lzma"
11
+ file_loc = "model_s.lzma"
12
+ if not Path(file_loc).exists():
13
+ raise Exception(f"File {file_loc} does not exist.")
14
+
15
+ with alive_bar(1, title="Loading model_s, takes up to ~2 minutes...", length=3) as progress_bar:
16
+ model_s = joblib.load(file_loc)
17
+ progress_bar() # pylint: disable=not-callable