From 441e2b6cce4b3730866dc63967fc23b6171baae4 Mon Sep 17 00:00:00 2001
From: Oisin Mac Aodha <omacaod@ed.ac.uk>
Date: Tue, 20 Dec 2022 16:53:57 +0000
Subject: [PATCH] rename outputs in app

---
 app.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/app.py b/app.py
index ca13f8e..ae44690 100644
--- a/app.py
+++ b/app.py
@@ -26,6 +26,7 @@ df =  gr.Dataframe(
         datatype=["str", "str", "str", "str"],
         row_count=1,
         col_count=(4, "fixed"),
+        label='Predictions'
     )
     
 examples = [['example_data/audio/20170701_213954-MYOMYS-LR_0_0.5.wav', 0.3],
@@ -96,7 +97,7 @@ gr.Interface(
     fn = make_prediction,
     inputs = [gr.Audio(source="upload", type="filepath", optional=True), 
               gr.Dropdown([0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9])],
-    outputs = [df, "image"],
+    outputs = [df, gr.Image(label="Visualisation")],
     theme = "huggingface",
     title = "BatDetect2 Demo",
     description = descr_txt,