Update app.py

app.py

@@ -46,7 +46,7 @@ def predict(img) -> Tuple[Dict, float]:
     start_time = timer()
     
     # Transform the target image and add a batch dimension
-    img = transform(img).unsqueeze(0)
+    img = transform(img).unsqueeze(dim=0)
     
     # Put model into evaluation mode and turn on inference mode
     TinyCNN_model.eval()