fix issues

app.py

@@ -50,8 +50,8 @@ def predict_image(image: Image.Image):
         # Format per-class probabilities
         prob_text = "\n".join([f"{class_names[i]}: {float(preds[i]):.2f}" for i in range(len(class_names))])
 
-        # Return formatted output
-        output = f"Prediction\nClass: {pred_class}\nConfidence: {confidence:.2f}\n\nPer-class probabilities\n{prob_text}"
+        # Return formatted output with titles
+        output = f"### Prediction Class: {pred_class}\n### Confidence: {confidence:.2f}\n### Per-class probabilities\n{prob_text}"
         return output
     except Exception as e:
         logger.error(f"Error during prediction: {str(e)}")