Update app.py

app.py

@@ -58,12 +58,12 @@ face_extractor = FaceExtractor(video_read_fn=video_read_fn,facedet=facedet)
 title = "FaceForensics++"
 
 def inference(vid):
-    return "./Labels/Fake.png", f"{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'])] )
     res = expit(faces_real_pred.mean())
     if res >= 0.5:
-        return "./Labels/Fake.jpg", f"{res*100:.2f}%"
+        return "./Labels/Fake.png", f"{res*100:.2f}%"
     else:
         return "./Labels/Real.jpg", f"{res*100:.2f}%"