Update app.py

app.py

@@ -19,7 +19,7 @@ warnings.filterwarnings('ignore')
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 sbcl = create_model("Weights/weights.tar")
-sbcl.to(device)
+sbcl.to(device) 
 
 face_detector = get_model("resnet50_2020-07-20", max_size=1024, device=device)
 face_detector.eval()
@@ -153,9 +153,9 @@ with gr.Blocks(title="Deepfake Detection CL", theme='upsatwal/mlsc_tiet', css=""
         gr.Examples(
             examples=examples,
             inputs=input_image,
-            outputs=output_image,
+            outputs=[label_probs, output_image],
             fn=predict_image,
-            cache_examples=False,
+            cache_examples=True,
         )
         btn.click(predict_image, inputs=input_image, outputs=[label_probs, output_image], api_name="/predict_image")
 
@@ -175,7 +175,7 @@ with gr.Blocks(title="Deepfake Detection CL", theme='upsatwal/mlsc_tiet', css=""
         gr.Examples(
             examples=examples_videos,
             inputs=input_video,
-            outputs=output_image_video,
+            outputs=[label_probs_video, output_image_video],
             fn=predict_video,
             cache_examples=True,
         )