Upload app.py

app.py

@@ -47,6 +47,7 @@ def load_model():
         logger.error(f"Error loading model: {e}")
         return f"Error loading model: {e}"
 
+@spaces.GPU
 def classify_solution(question: str, solution: str):
     """
     Classify the math solution