Update model/mistral_7b_ielts_evaluator.py

model/mistral_7b_ielts_evaluator.py

@@ -2,9 +2,10 @@ from typing import *
 import torch
 from transformers import pipeline
 import spaces
-from .common import Grader, device
+from .common import Grader
 
 model_name = "JacobLinCool/mistral-7b-ielts-evaluator-safetensors"
+device = -1  # Chỉ sử dụng CPU
 pipe = pipeline("text-generation", model_name, device=device)
 
 system = """In this task, you are required to evaluate an IELTS Writing Task 2 essay. Consider the following four criteria and provide a detailed assessment for each, along with a suggested band score:
@@ -48,7 +49,6 @@ class Mistral_7b_IELTS_Evaluator(Grader):
     def info(self) -> str:
         return "Safetensors version of [chillies/mistral-7b-ielts-evaluator-q4](https://huggingface.co/chillies/mistral-7b-ielts-evaluator-q4)"
 
-    @spaces.GPU(duration=120)
     @torch.no_grad()
     def grade(self, question: str, answer: str) -> Tuple[float, str]:
         text = f"""{system}