engrharis commited on
Commit
70845c6
·
verified ·
1 Parent(s): 687db18

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
+ import torch
4
+
5
+ # Load the model and tokenizer
6
+ tokenizer = AutoTokenizer.from_pretrained("Hate-speech-CNERG/urdu-codemixed-abusive-MuRIL")
7
+ model = AutoModelForSequenceClassification.from_pretrained("Hate-speech-CNERG/urdu-codemixed-abusive-MuRIL")
8
+
9
+ # Define labels in Urdu
10
+ labels = {0: "نارمل (معمول)", 1: "گالی گلوچ (بدتمیزی)"}
11
+
12
+ # App title and description
13
+ st.title("اردو متن کا تجزیہ کریں")
14
+ st.write("یہ ایپ آپ کے فراہم کردہ اردو متن کی نوعیت (نارمل یا گالی گلوچ) کو پہچانتی ہے۔")
15
+
16
+ # User input
17
+ user_input = st.text_area("اردو متن درج کریں:")
18
+
19
+ if st.button("تجزیہ کریں"):
20
+ if user_input.strip():
21
+ # Tokenize and classify the input text
22
+ inputs = tokenizer(user_input, return_tensors="pt", truncation=True, padding=True)
23
+ outputs = model(**inputs)
24
+ logits = outputs.logits
25
+ predicted_class = torch.argmax(logits, dim=1).item()
26
+
27
+ # Display the result
28
+ st.write(f"متن کی نوعیت: **{labels[predicted_class]}**")
29
+ else:
30
+ st.warning("براہ کرم متن درج کریں!")