change examples

app.py

@@ -90,13 +90,11 @@ def process_example(image_path):
     model_type = determine_model_type(image_path)
     return Image.open(image_path), model_type
 
-
-example_images = [
-    "examples/fake/facial.jpg",
-    "examples/fake/general.jpg",
-    "examples/real/facial.jpg",
-    "examples/real/general.jpg",
-]
+fake_examples, real_examples = [], []
+for example in os.listdir("examples/fake"):
+    fake_examples.append(os.path.join("examples/fake", example))
+for example in os.listdir("examples/real"):
+    real_examples.append(os.path.join("examples/real", example))
 
 with gr.Blocks() as demo:
   
@@ -105,22 +103,34 @@ with gr.Blocks() as demo:
     input_type = gr.Radio(["Image", "Video"], label="Choose Input Type", value="Image")
 
     model_type = gr.Radio(["Facial", "General"], label="Choose Model Type", value="General")
-  
-    image_input = gr.Image(type="pil", label="Upload Image", visible=True)
-    video_input = gr.Video(label="Upload Video", visible=False)
-  
+
+    H, W = 300, 300
+    image_input = gr.Image(type="pil", label="Upload Image", visible=True, height=H, width=W)
+    video_input = gr.Video(label="Upload Video", visible=False, height=H, width=W)
+
     process_button = gr.Button("Run Model")
 
-    pred_score_output = gr.Textbox(label="Prediction Score")
-    attn_map_output = gr.Image(type="pil", label="Attention Map")
+    pred_score_output = gr.Textbox(label="Prediction Score : 0 - REAL, 1 - FAKE")
+    attn_map_output = gr.Image(type="pil", label="Attention Map", height=H, width=W)
 
     # Example Images 추가
     gr.Examples(
-        examples=example_images,
+        examples=fake_examples,
+        inputs=[image_input],
+        outputs=[image_input, model_type],
+        fn=process_example,
+        cache_examples=False,
+        examples_per_page=10,
+        label="Fake Examples"
+    )
+    gr.Examples(
+        examples=real_examples,
         inputs=[image_input],
         outputs=[image_input, model_type],
         fn=process_example,
-        cache_examples=False
+        cache_examples=False,
+        examples_per_page=10,
+        label="Real Examples"
     )
   
     input_type.change(fn=change_input, inputs=[input_type], outputs=[image_input, video_input])