minor bug fix!

app.py

@@ -10,7 +10,8 @@ from torchvision.transforms import v2
 
 from model import MAE_ViT, MAE_Encoder, MAE_Decoder, MAE_Encoder_FeatureExtractor
 
-path  = [['images/cat.jpg'], ['images/dog.jpg'], ['images/horse.jpg'], ['images/airplane.jpg'], ['images/truck.jpg']]
+path_1  = [['images/cat.jpg'], ['images/dog.jpg'], ['images/horse.jpg'], ['images/airplane.jpg'], ['images/truck.jpg']]
+path_2  = [['images/cat.jpg'], ['images/dog.jpg'], ['images/horse.jpg'], ['images/airplane.jpg'], ['images/truck.jpg']]
 device = torch.device("cpu")
 
 model_name = "model/no_mode/vit-t-mae-pretrain.pt"
@@ -128,19 +129,27 @@ def visualize_single_image_pca_mode(image_path):
 
     return np.array(plt.imread("output.png"))
 
-inputs_image = [
+inputs_image_1 = [
     gr.components.Image(type="filepath", label="Input Image"),
 ]
 
-outputs_image = [
+outputs_image_1 = [
+    gr.components.Image(type="numpy", label="Output Image"),
+]
+
+inputs_image_2 = [
+    gr.components.Image(type="filepath", label="Input Image"),
+]
+
+outputs_image_2 = [
     gr.components.Image(type="numpy", label="Output Image"),
 ]
 
 inference_no_mode = gr.Interface(
     fn=visualize_single_image_no_mode,
-    inputs=inputs_image,
-    outputs=outputs_image,
-    examples=path,
+    inputs=inputs_image_1,
+    outputs=outputs_image_1,
+    examples=path_1,
     cache_examples = False,
     title="MAE-ViT Image Reconstruction",
     description="This is a demo of the MAE-ViT model for image reconstruction.",
@@ -148,9 +157,9 @@ inference_no_mode = gr.Interface(
 
 inference_pca_mode = gr.Interface(
     fn=visualize_single_image_pca_mode,
-    inputs=inputs_image,
-    outputs=outputs_image,
-    examples=path,
+    inputs=inputs_image_2,
+    outputs=outputs_image_2,
+    examples=path_2,
     title="MAE-ViT Image Reconstruction",
     description="This is a demo of the MAE-ViT model for image reconstruction.",
 )