Update pipeline.py

pipeline.py

@@ -132,7 +132,9 @@ def deepfakes_video_predict(input_video):
     for face in faces:
 
         face2 = face/255
-        pred = model.predict(np.expand_dims(face2, axis=0))[0]
+        pred = model(np.expand_dims(face2, axis=0))
+        pred = list(pred.values())[0].numpy()[0]
+
         real, fake = pred[0], pred[1]
         real_res.append(real)
         fake_res.append(fake)
@@ -162,7 +164,9 @@ def deepfakes_video_predict(input_video):
 def deepfakes_image_predict(input_image):
     faces = detection_image_pipeline(input_image)
     face2 = faces/255
-    pred = model.predict(np.expand_dims(face2, axis = 0))[0]
+    pred = model(np.expand_dims(face2, axis=0))
+    pred = list(pred.values())[0].numpy()[0]
+
     real, fake = pred[0], pred[1]
     if real > 0.5:
         text2 = "The image is REAL. \n Deepfakes Confidence: " + str(round(100 - (real*100), 3)) + "%"