Update app.py

app.py

@@ -3,46 +3,42 @@ from transformers import pipeline
 import requests
 import spacy
 
-# Load models
-nlp = spacy.load("en_core_web_sm")  # For coherence analysis
-vocab_model = pipeline("text-classification", model="textattack/bert-base-uncased-CoLA")  # Better for essays
+# Load free models
+nlp = spacy.load("en_core_web_sm")
+vocab_model = pipeline("text-classification", model="roberta-base")  # Free alternative
 
 def evaluate_ielts(essay):
     # Vocabulary Assessment
-    vocab_result = vocab_model(essay)[0]
-    vocab_score = round(vocab_result['score'] * 9 * 1.2, 1)  # Adjusted multiplier
+    try:
+        vocab_score = min(9, vocab_model(essay)[0]['score'] * 9 + 1)  # Adjusted scaling
+    except:
+        vocab_score = 6.0
     
-    # Grammar Check (LanguageTool API)
+    # Grammar Check (Free API)
     grammar_response = requests.post(
         "https://api.languagetool.org/v2/check",
         data={"text": essay, "language": "en-GB"}
     )
-    grammar_errors = len(grammar_response.json()['matches'])
+    grammar_errors = len(grammar_response.json().get('matches', []))
     grammar_score = max(1, 9 - (grammar_errors // 1.5))  # 1.5 errors = 1 point deduction
     
     # Coherence Analysis
     doc = nlp(essay)
-    transition_words = ["however", "moreover", "therefore", "consequently"]
+    transition_words = ["however", "moreover", "therefore", "furthermore"]
     transitions = sum(1 for token in doc if token.text.lower() in transition_words)
     coherence_score = min(9, transitions + 4)  # 4-9 scale
     
-    # Overall Band (IELTS-style weighting)
+    # Overall Band Calculation
     overall = (vocab_score * 0.4) + (grammar_score * 0.3) + (coherence_score * 0.3)
     
     return {
-        "Vocabulary Band": min(9, vocab_score),
+        "Vocabulary Band": round(vocab_score, 1),
         "Grammar Band": grammar_score,
         "Coherence Band": coherence_score,
         "Overall Band": round(overall, 1)
     }
 
+# Create Interface
 gr.Interface(
     fn=evaluate_ielts,
-    inputs=gr.Textbox(label="Essay", lines=7),
-    outputs=gr.JSON(),
-    examples=[[
-        "Education plays a vital role in societal development. " 
-        "Firstly, it equips individuals with critical thinking skills. " 
-        "Furthermore, educated citizens contribute to economic growth."
-    ]]
-).launch()
+    inputs=gr.Textbox(label="Paste Your Essay",