Spaces:

rahbi
/

alclam-dialect-identification

Sleeping

rahbi commited on Jul 25, 2024

Commit

9d01d40

verified ·

1 Parent(s): 38a94db

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-id")
 # Define the prediction function
 def classify_text(text):

 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):