serhii-korobchenko commited on
Commit
f7fe391
·
verified ·
1 Parent(s): 468a15a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ model.load_state_dict(state_dict, strict=False)
27
  model.eval()
28
 
29
  def predict(im):
30
- print (im)
31
  x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.
32
 
33
  with torch.no_grad():
 
27
  model.eval()
28
 
29
  def predict(im):
30
+ print (len(im))
31
  x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.
32
 
33
  with torch.no_grad():