Update app.py

app.py

@@ -3,22 +3,23 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
 # Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("Hate-speech-CNERG/urdu-codemixed-abusive-MuRIL")
-model = AutoModelForSequenceClassification.from_pretrained("Hate-speech-CNERG/urdu-codemixed-abusive-MuRIL")
+model_name = "Hate-speech-CNERG/urdu-abusive-MuRIL"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
 # Define labels in Urdu
 labels = {0: "نارمل (معمول)", 1: "گالی گلوچ (بدتمیزی)"}
 
-# App title and description
-st.title("اردو متن کا تجزیہ کریں")
-st.write("یہ ایپ آپ کے فراہم کردہ اردو متن کی نوعیت (نارمل یا گالی گلوچ) کو پہچانتی ہے۔")
+# Streamlit UI
+st.title("اردو بدتمیزی کا تجزیہ")
+st.write("یہ ایپ آپ کے متن کا تجزیہ کرے گی اور بتائے گی کہ یہ نارمل ہے یا بدتمیز۔")
 
-# User input
-user_input = st.text_area("اردو متن درج کریں:")
+# Input text from user
+user_input = st.text_area("اپنا متن یہاں درج کریں:")
 
 if st.button("تجزیہ کریں"):
     if user_input.strip():
-        # Tokenize and classify the input text
+        # Tokenize and classify the input
         inputs = tokenizer(user_input, return_tensors="pt", truncation=True, padding=True)
         outputs = model(**inputs)
         logits = outputs.logits