output info

interfaces/cap_minor_media.py

@@ -190,6 +190,8 @@ def predict_flat(text, model_id, tokenizer_id, HF_TOKEN=None):
     This method returns predictions made by a single model. Both media codes and minor topics may appear in the output list. **Only the top 10 most confident labels are displayed**.
     """
     
+    output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'
+    
     return interpretation_info, output_pred, output_info