ciivam commited on
Commit
51e303d
·
verified ·
1 Parent(s): 3e6edef

Update Deepfake_detection.ipynb

Browse files
Files changed (1) hide show
  1. 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.85 else \"fake\"\n",
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",