Update app.py

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],
@@ -90,16 +91,18 @@ def generate_results_image(audio_file, anns):
 descr_txt = "Demo of BatDetect2 deep learning-based bat echolocation call detection. " \
           "<br>This model is only trained on bat species from the UK. If the input " \
           "file is longer than 2 seconds, only the first 2 seconds will be processed." \
-          "<br>For more information, check out the [paper](https://www.biorxiv.org/content/10.1101/2022.12.14.520490v1)."
+          "<br>Check out the paper [here](https://www.biorxiv.org/content/10.1101/2022.12.14.520490v1)."
 
 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,
     examples = examples,
     allow_flagging = 'never',
 ).launch()
+
+