Update mmlu_eval.py

mmlu_eval.py

@@ -3,6 +3,7 @@ import random
 import evaluate
 from datasets import load_dataset
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import spaces
 
 # Load Accuracy Metric
 accuracy_metric = evaluate.load("accuracy")
@@ -10,6 +11,7 @@ accuracy_metric = evaluate.load("accuracy")
 # Load MMLU dataset
 mmlu_dataset = load_dataset("cais/mmlu", "all")
 
+@spaces.GPU
 def generate_answer(model, tokenizer, question):
     """
     Generates an answer using Mistral's instruction format.