Update app.py

app.py

@@ -111,6 +111,15 @@ def predict(cfg, sequence):
     predictions = list(itertools.chain.from_iterable(predictions))
     outputs["raw prediction values"] = predictions
     outputs["binary prediction values"] = [1 if x > 0.5 else 0 for x in predictions]
+
+    html_output = f"""
+    <div style='border: 2px solid #4CAF50; padding: 10px; border-radius: 10px;'>
+        <h3 style='color: #4CAF50;'>Protein: {cfg.sequence_col}</h3>
+        <p><strong>Stability:</strong> {outputs['raw prediction values']}</p>
+        <p><strong>Organism:</strong> {outputs['binary prediction values']}</p>
+    </div>
+    """
+    
     return outputs