Spaces:

rahbi
/

alclam-dialect-identification

Sleeping

rahbi commited on Jul 26, 2024

Commit

0bd1b52

verified ·

1 Parent(s): b16e1f8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 # Initialize the classifier pipeline
-classifier = pipeline("text-classification", model="rahbi/alclam-base-v1-dialect")
 # Define the prediction function
 def classify_text(text):

 from transformers import pipeline
 # Initialize the classifier pipeline
+classifier = pipeline("text-classification", model="rahbi/alclam-base-v2-DID")
 # Define the prediction function
 def classify_text(text):