Update app.py

app.py

@@ -1,4 +1,17 @@
 import streamlit as st
+from transformers import pipeline
 
+# Load the model
+model = pipeline("text-classification", model="Alshargi/arabic-msa-dialects-segmentation")
+
+# Slider to select a value
 x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+
+# Display the squared value
+st.write(x, 'squared is', x * x)
+
+# Make prediction using the loaded model
+prediction = model(x)
+
+# Display the prediction
+st.write("Prediction:", prediction)