vishalbakshi commited on
Commit
aad5d7e
1 Parent(s): ab9d8b6

retrain model2.pkl with old named getters

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. model2.pkl +1 -1
app.py CHANGED
@@ -19,13 +19,13 @@ app.add_middleware(
19
 
20
  def get_x(i):
21
  # Convert NumPy array to a single-channel PIL image with inverted colors
22
- return PILImageBW.create(x[i])
23
 
24
  def get_y(i):
25
- return y[i].astype(np.float32)
26
 
27
  def get_items(_):
28
- return range(len(x))
29
 
30
  # Load the model
31
  #learn = load_learner('model.pkl')
 
19
 
20
  def get_x(i):
21
  # Convert NumPy array to a single-channel PIL image with inverted colors
22
+ return PILImageBW.create(all_noise[i])
23
 
24
  def get_y(i):
25
+ return all_thresh[i].astype(np.float32)
26
 
27
  def get_items(_):
28
+ return range(len(all_noise))
29
 
30
  # Load the model
31
  #learn = load_learner('model.pkl')
model2.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1782bbc0c2083d506c1eb98179a584a19a66964286be72acc30685b526522508
3
  size 88336243
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a939f9cd41b8f2639c279bebd6544a4decc38992dd75fcb3ae5d819c274942a
3
  size 88336243