UI polish: logo, human-readable reliability, default ubiquitin sequence

app.py

@@ -132,16 +132,35 @@ def predict(sequence: str):
     pred = pred_texts[0]
     r = float(r_pred.cpu().tolist()[0] if torch.is_tensor(r_pred) else r_pred[0])
     p1 = float(r_prob_class1.cpu().tolist()[0] if torch.is_tensor(r_prob_class1) else r_prob_class1[0])
-    return pred, f"{r:.4f}", f"{p1:.4f}"
+    return pred, format_reliability(r), f"{p1:.4f}"
 
 
 EXAMPLE_SEQ = (
-    "MKTVRQERLKSIVRILERSKEPVSGAQLAEELSVSRQVIVQDIAYLRSLGYNIVATPRGYVLAGG"
+    "MQIFVKTLTGKTITLEVEPSDTIENVKAKIQDKEGIPPDQQRLIFAGKQLEDGRTLSDYNIQKESTLHLVLRLRGG"
 )
 
+RELIABILITY_LABELS = {
+    1.0: "High-confidence correct",
+    0.0: "High-confidence incorrect",
+    0.5: "Uncertain",
+    -1.0: "Vague",
+}
+
+
+def format_reliability(r: float) -> str:
+    label = RELIABILITY_LABELS.get(r, "Unknown")
+    return f"{label} (class={r:g})"
+
+
 with gr.Blocks(title="ProtTale") as demo:
+    gr.Image(
+        value="prottale_logo.png",
+        show_label=False,
+        show_download_button=False,
+        container=False,
+        height=120,
+    )
     gr.Markdown(
-        "# ProtTale\n"
         "Protein amino-acid sequence → Swiss-Prot-style function description, "
         "with a reliability score for the generated text.\n\n"
         "**Note:** this Space runs on CPU; a single beam-3 generation typically takes ~30–120 s."
@@ -151,8 +170,8 @@ with gr.Blocks(title="ProtTale") as demo:
     run_btn = gr.Button("Predict", variant="primary")
     with gr.Column():
         pred_out = gr.Textbox(label="Predicted function")
-        r_out = gr.Textbox(label="Reliability class (−1 / 0 / 0.5 / 1)")
-        p_out = gr.Textbox(label="P(reliability = 1.0)")
+        r_out = gr.Textbox(label="Reliability")
+        p_out = gr.Textbox(label="P(high-confidence correct)")
 
     run_btn.click(predict, inputs=[seq_in], outputs=[pred_out, r_out, p_out])