Rawan-Otaibi commited on
Commit
c827c33
·
verified ·
1 Parent(s): ce1b944

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +29 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import torch
3
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
+
5
+ # تحميل النموذج و Tokenizer
6
+ model_name = "aubmindlab/bert-base-arabertv02"
7
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
8
+ model = AutoModelForSequenceClassification.from_pretrained("model")
9
+
10
+ # التصنيفات
11
+ labels = ["جبر", "تفاضل", "تكامل", "هندسة", "إحصاء"]
12
+
13
+ st.set_page_config(page_title="روان العتيبي | تصنيف مسائل الرياضيات")
14
+ st.title("تصنيف مسائل الرياضيات")
15
+ st.markdown("أدخل مسألة رياضية وسيقوم النموذج بتحديد نوعها (جبر، تفاضل، ...).")
16
+
17
+ # إدخال المستخدم
18
+ text = st.text_area("أدخل المسألة هنا:")
19
+
20
+ # زر التصنيف
21
+ if st.button("صنف"):
22
+ if not text.strip():
23
+ st.warning("يرجى إدخال مسألة.")
24
+ else:
25
+ inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
26
+ with torch.no_grad():
27
+ outputs = model(**inputs)
28
+ pred = torch.argmax(outputs.logits, dim=1).item()
29
+ st.success(f"التصنيف: **{labels[pred]}**")
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ streamlit
2
+ transformers
3
+ torch