Changing to fitness message classifier

app.py

@@ -1,10 +1,17 @@
 import streamlit as st
 from transformers import pipeline
-import torch
 
-pipe = pipeline(model='DicksonMassawe/finetuning-spam-detection-model')
+# Load pre-trained model pipeline
+model_name = "lucieackley/autotrain-fitness_message_classification-40235104670"
+pipe = pipeline('text-classification', model=model_name)
+
+st.title("Fitness Message Classifier")
+
 text = st.text_area("Enter text: ")
+
 if text:
+    # Get model's prediction
     out = pipe(text)
-    st.json(out)
 
+    # Show prediction
+    st.write(f"The text was classified as: {out[0]['label']} with a score of {out[0]['score']:.2f}")