doevent commited on
Commit
3ccf46a
1 Parent(s): 0f5662c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -21,6 +21,7 @@ def inference(image, size):
21
  result = model8.predict(image.convert('RGB'))
22
  if torch.cuda.is_available():
23
  torch.cuda.empty_cache()
 
24
  return result
25
 
26
 
 
21
  result = model8.predict(image.convert('RGB'))
22
  if torch.cuda.is_available():
23
  torch.cuda.empty_cache()
24
+ print(f"Image size: {size} ... OK")
25
  return result
26
 
27