Sahal Mulki commited on
Commit
58a5eb9
·
1 Parent(s): 2f2d1d1
Files changed (2) hide show
  1. SpottingDiffusion(1).keras +3 -0
  2. app.py +2 -2
SpottingDiffusion(1).keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db468f1e52a7ae3a4c4fdb93c79088732f4c23e7b85393b63a4572c1335fb7d3
3
+ size 261293289
app.py CHANGED
@@ -28,8 +28,8 @@ custom_objects = {} # Use an empty dict if you don't have any
28
 
29
 
30
 
31
- model = tf.keras.models.load_model("pretrained/spottingdiffusion.keras")
32
- use_hack = True
33
  if use_hack:
34
  for variable in model.trainable_weights:
35
  variable.regularizer = None
 
28
 
29
 
30
 
31
+ model = tf.keras.models.load_model("SpottingDiffusion(1).keras")
32
+ use_hack = False
33
  if use_hack:
34
  for variable in model.trainable_weights:
35
  variable.regularizer = None