Update app.py

app.py

@@ -61,6 +61,8 @@ def inference(vid):
     #return "./Labels/Fake.png", f"{vid}"
     vid_real_faces = face_extractor.process_video(vid)
     faces_real_t = torch.stack( [ transf(image=frame['faces'][0])['image'] for frame in vid_real_faces if len(frame['faces'])] )
+    with torch.no_grad():
+        faces_real_pred = net(faces_real_t.to(device)).cpu().numpy().flatten()
     res = expit(faces_real_pred.mean())
     if res >= 0.5:
         return "./Labels/Fake.png", f"{res*100:.2f}%"