Spaces:
Running
Running
Update Deepfake_detection.ipynb
Browse files- Deepfake_detection.ipynb +1 -1
Deepfake_detection.ipynb
CHANGED
@@ -995,7 +995,7 @@
|
|
995 |
"\n",
|
996 |
" with torch.no_grad():\n",
|
997 |
" output = torch.sigmoid(model(face).squeeze(0))\n",
|
998 |
-
" prediction = \"real\" if output.item() < 0.
|
999 |
" \n",
|
1000 |
" real_prediction = 1 - output.item()\n",
|
1001 |
" fake_prediction = output.item()\n",
|
|
|
995 |
"\n",
|
996 |
" with torch.no_grad():\n",
|
997 |
" output = torch.sigmoid(model(face).squeeze(0))\n",
|
998 |
+
" prediction = \"real\" if output.item() < 0.5 else \"fake\"\n",
|
999 |
" \n",
|
1000 |
" real_prediction = 1 - output.item()\n",
|
1001 |
" fake_prediction = output.item()\n",
|