update submit

src/submission/submit.py

@@ -18,6 +18,8 @@ from src.submission.check_validity import (
     is_model_on_hub,
 )
 
++import spaces
+
 REQUESTED_MODELS = None
 USERS_TO_SUBMISSION_DATES = None
 
@@ -66,6 +68,7 @@ def get_top_prediction(text, tokenizer, model):
     top_option = max(option_logits, key=lambda x: x[0])[1]
     return top_option
 
+@spaces.GPU(duration=300)
 def evaluate_model_accuracy_by_subject(model_name, num_examples):
     try:
         # Load the model and tokenizer