Update app.py

app.py

@@ -1,22 +1,22 @@
-import gradio as gr
-from transformers import pipeline
-
-# Initialize the classifier pipeline
-classifier = pipeline("text-classification", model="rahbi/alclam-base-v1-DID")
-
-# Define the prediction function
-def classify_text(text):
-    result = classifier(text)
-    return result
-
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=classify_text,
-    inputs="text",
-    outputs="json",
-    title="Arabic Text Classifier",
-    description="Classify Arabic text using the rahbi/alclam-base-v1-DID model."
-)
-
-# Launch the interface
+import gradio as gr
+from transformers import pipeline
+
+# Initialize the classifier pipeline
+classifier = pipeline("text-classification", model="rahbi/alclam-base-v1-dialect-id")
+
+# Define the prediction function
+def classify_text(text):
+    result = classifier(text)
+    return result
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=classify_text,
+    inputs="text",
+    outputs="json",
+    title="Arabic Text Classifier",
+    description="Classify Arabic text using the rahbi/alclam-base-v1-DID model."
+)
+
+# Launch the interface
 iface.launch()