feat: gradio app

README.md

@@ -1,6 +1,6 @@
 ---
 title: EssayScoring
-emoji: 📈
+emoji: 📝
 colorFrom: gray
 colorTo: blue
 sdk: gradio
@@ -8,6 +8,8 @@ sdk_version: 4.40.0
 app_file: app.py
 pinned: false
 license: mit
+preload_from_hub:
+    - JacobLinCool/IELTS_essay_scoring_safetensors
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,88 @@
+from typing import *
+import gradio as gr
+import numpy as np
+import spaces
+import torch
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+
+model_name = "JacobLinCool/IELTS_essay_scoring_safetensors"
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+
+@spaces.GPU
+def grade(question: str, answer: str) -> Tuple[float, float, float, float, float]:
+    if len(question) < 30 or len(answer) < 30:
+        raise gr.Error("Please enter more than 30 characters")
+
+    text = f"{question} {answer}"
+
+    inputs = tokenizer(
+        text, return_tensors="pt", padding=True, truncation=True, max_length=512
+    )
+    with torch.no_grad():
+        outputs = model(**inputs)
+        predictions = outputs.logits.squeeze()
+
+    predicted_scores = predictions.numpy()
+    normalized_scores = (predicted_scores / predicted_scores.max()) * 9
+    rounded_scores = np.round(normalized_scores * 2) / 2
+
+    return tuple(rounded_scores)
+
+
+with gr.Blocks() as app:
+    gr.Markdown("# Essay Scoring")
+
+    with gr.Row():
+        with gr.Column():
+            question = gr.Textbox(
+                label="Enter the question here",
+                placeholder="Write the question here",
+                lines=3,
+            )
+            essay = gr.Textbox(
+                label="Enter your essay here",
+                placeholder="Write your essay here",
+                lines=10,
+            )
+            btn = gr.Button("Grade Essay", variant="primary")
+        with gr.Column():
+            task_achievement = gr.Number(label="Task Achievement")
+            coherence_cohesion = gr.Number(label="Coherence and Cohesion")
+            vocabulary = gr.Number(label="Vocabulary")
+            grammar = gr.Number(label="Grammar")
+            overall = gr.Number(label="Overall")
+
+    btn.click(
+        fn=grade,
+        inputs=[question, essay],
+        outputs=[task_achievement, coherence_cohesion, vocabulary, grammar, overall],
+    )
+
+    gr.Examples(
+        [
+            [
+                "It is important for all towns and cities to have large public spaces such as squares and parks. Do you agree or disagree with this statement?",
+                (
+                    "It is crucial for all metropolitan cities and towns to "
+                    "have some recreational facilities like parks and squares because of their numerous benefits. A number of "
+                    "arguments surround my opinion, and I will discuss it in upcoming paragraphs. To commence with, the first "
+                    "and the foremost merit is that it is beneficial for the health of people because in morning time they can "
+                    "go for walking as well as in the evenings, also older people can spend their free time with their loved ones, "
+                    "and they can discuss about their daily happenings. In addition, young people do lot of exercise in parks and "
+                    "gardens to keep their health fit and healthy, otherwise if there is no park they glue with electronic gadgets "
+                    "like mobile phones and computers and many more. Furthermore, little children get best place to play, they play "
+                    "with their friends in parks if any garden or square is not available for kids then they use roads and streets "
+                    "for playing it can lead to serious incidents. Moreover, parks have some educational value too, in schools, "
+                    "students learn about environment protection in their studies and teachers can take their pupils to parks because "
+                    "students can see those pictures so lively which they see in their school books and they know about importance "
+                    "and protection of trees and flowers. In recapitulate, parks holds immense importance regarding education, health "
+                    "for people of every society, so government should build parks in every city and town."
+                ),
+            ],
+        ],
+        inputs=[question, essay],
+    )
+
+app.launch()

requirements.txt

@@ -0,0 +1,6 @@
+torch
+transformers
+accelerate
+spaces
+gradio
+numpy