Rawan-Otaibi commited on
Commit
9ad871c
·
verified ·
1 Parent(s): 5569fd2

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -29
app.py DELETED
@@ -1,29 +0,0 @@
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]}**")