Hali5 commited on
Commit
7d56ac1
·
1 Parent(s): 8489b61

fixed again

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -44,9 +44,8 @@ def predict(image):
44
  return None
45
 
46
  img_resized = image.resize((64, 64), Image.Resampling.LANCZOS)
47
- pil_img = Image.fromarray(img_resized.astype('uint8'), 'RGB')
48
-
49
- img_tensor = tf(pil_img).unsqueeze(0).to(device)
50
 
51
  with torch.no_grad():
52
  outputs = model(img_tensor)
 
44
  return None
45
 
46
  img_resized = image.resize((64, 64), Image.Resampling.LANCZOS)
47
+
48
+ img_tensor = tf(img_resized).unsqueeze(0).to(device)
 
49
 
50
  with torch.no_grad():
51
  outputs = model(img_tensor)