Create app.py

app.py

@@ -0,0 +1,28 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+import torch.nn.functional as F
+
+app = FastAPI(title="Arabic Readability API")
+
+class InputText(BaseModel):
+    text: str
+
+MODEL_NAME = "aubmindlab/bert-base-arabertv2"  # أو ضعي رابط موديلك الخاص
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
+model.eval()
+
+@app.get("/")
+def home():
+    return {"message": "Arabic Readability API is running!"}
+
+@app.post("/predict")
+def predict(inp: InputText):
+    inputs = tokenizer(inp.text, return_tensors="pt", truncation=True, padding=True, max_length=512)
+    with torch.no_grad():
+        outputs = model(**inputs)
+        probs = F.softmax(outputs.logits, dim=-1)
+        pred = int(torch.argmax(probs))
+    return {"label": pred, "confidence": float(probs[0][pred])}