This commit is contained in:
Yousif Mansour 2024-07-17 15:33:42 -04:00 committed by GitHub
commit 367595c30a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -758,8 +758,7 @@
}
],
"source": [
"img = PILImage.create(uploader.data[0])\n",
"is_cat,_,probs = learn.predict(img)\n",
"is_cat,_,probs = learn.predict(uploader.data[0])\n",
"print(f\"Is this a cat?: {is_cat}.\")\n",
"print(f\"Probability it's a cat: {probs[1].item():.6f}\")"
]