Update app.py

app.py

@@ -79,9 +79,12 @@ def prepare_text(text):
 
 
 
-# Load the model from the local file
-model_path = "arabic-msa-dialects-segmentation-v1.pkl"
-model = pipeline("text-classification", model=model_path)
+
+from transformers import AutoModel
+
+# Load the model using the Hugging Face model hub
+model = AutoModel.from_pretrained("Alshargi/arabic-msa-dialects-segmentation")
+