Update README.md

README.md

@@ -68,4 +68,4 @@ with torch.no_grad():
     temp, predicted = torch.max(outputs, 1)
 
 # Output the prediction
-print(f'The predicted class for the image is: {"Cat" if predicted.item() == 0 else "Dog"}')
+print(f'The predicted class for the image is: {"Cat" if predicted.item() == 0 else "Dog"}')