Update app.py

app.py

@@ -35,14 +35,14 @@ def deepfakeclassifier(potential_test_video, option):
     models = []
     model_paths = [os.path.join(weights_dir, model) for model in models_dir]
     for path in model_paths:
-        model = DeepFakeClassifier(encoder="tf_efficientnet_b7_ns").to('cuda')
+        model = DeepFakeClassifier(encoder="tf_efficientnet_b7_ns").to('cpu')
         print("loading state dict {}".format(path))
         checkpoint = torch.load(path, map_location="cpu")
         state_dict = checkpoint.get("state_dict", checkpoint)
         model.load_state_dict({re.sub("^module.", "", k): v for k, v in state_dict.items()}, strict=True)
         model.eval()
         del checkpoint
-        models.append(model.half())
+        models.append(model.float())
 
     frames_per_video = 32
     video_reader = VideoReader()