using strings as input/outputs

app.py

@@ -92,6 +92,9 @@ def NormalizeData(data):
 
 # Visualization
 def visualize_similarities(image1, image2):
+    image1 = Image.fromarray(image1)
+    image2 = Image.fromarray(image2)
+
     a = get_layer4(image1).squeeze()
     b = get_layer4(image2).squeeze()
     sim1, sim2 = compute_spatial_similarity(a, b)
@@ -127,12 +130,9 @@ def visualize_similarities(image1, image2):
 # GRADIO APP
 main = gr.Interface(
     fn=visualize_similarities,
-    inputs=[
-        gr.Image(type="pil"),
-        gr.Image(type="pil"),
-    ],
+    inputs=["image", "image"],
     allow_flagging="never",
-    outputs=gr.Plot(type="matplotlib"),
+    outputs="plot",
     cache_examples=True,
     enable_queue=False,
     examples=[