Update app.py

app.py

@@ -1,8 +1,11 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, AutoConfig
 
-# Load the model
-model = pipeline("text-classification", model="Alshargi/arabic-msa-dialects-segmentation")
+# Load the model configuration from config.json
+config = AutoConfig.from_pretrained("Alshargi/arabic-msa-dialects-segmentation")
+
+# Load the model using the configuration
+model = pipeline("text-classification", model="Alshargi/arabic-msa-dialects-segmentation", config=config)
 
 # Slider to select a value
 x = st.slider('Select a value')