File size: 2,138 Bytes
845a5d4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | import gradio as gr
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
import torch
# ----------------------------
# Model setup (lightweight & free)
# ----------------------------
MODEL_ID = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
model = AutoModelForCausalLM.from_pretrained(
MODEL_ID,
dtype=torch.float32,
device_map="auto"
)
pipe = pipeline(
"text-generation",
model=model,
tokenizer=tokenizer
)
# ----------------------------
# IELTS Scoring Function
# ----------------------------
def score_ielts_writing(text):
prompt = f"""
You are a certified IELTS Writing examiner.
Ignore the essay content in your output.
Score the essay strictly using IELTS band descriptors.
Output ONLY the final scores in JSON format.
Do NOT repeat the essay.
Score these criteria:
- Task Response (TR)
- Coherence and Cohesion (CC)
- Lexical Resource (LR)
- Grammatical Range and Accuracy (GRA)
MANDATORY JSON FORMAT:
{{
"task_response": 0.0,
"coherence_cohesion": 0.0,
"lexical_resource": 0.0,
"grammatical_range_accuracy": 0.0,
"overall_band": 0.0,
"feedback": "2–3 sentence examiner feedback"
}}
ESSAY (QUESTION + ESSAY):
{text}
Now, output ONLY the JSON scores as above.
"""
output = pipe(
prompt,
max_new_tokens=450, # enough for 300-word essay + JSON
temperature=0.3, # deterministic scoring
do_sample=False,
repetition_penalty=1.05
)[0]["generated_text"]
# remove the prompt part from the output
return output[len(prompt):].strip()
# ----------------------------
# Gradio UI
# ----------------------------
iface = gr.Interface(
fn=score_ielts_writing,
inputs=gr.Textbox(
lines=18,
label="Paste IELTS QUESTION + ESSAY",
placeholder="QUESTION:\n...\n\nESSAY:\n..."
),
outputs=gr.Textbox(label="IELTS Band Scores (JSON)"),
title="IELTS Writing AI Scorer (Free • CPU • Fast)",
description="Paste the IELTS Writing Task 2 question followed by the essay. Returns band scores + examiner feedback."
)
iface.launch()
|