update submit
Browse files- src/submission/submit.py +1 -1
src/submission/submit.py
CHANGED
@@ -68,7 +68,7 @@ def get_top_prediction(text, tokenizer, model):
|
|
68 |
top_option = max(option_logits, key=lambda x: x[0])[1]
|
69 |
return top_option
|
70 |
|
71 |
-
@spaces.GPU(duration=
|
72 |
def evaluate_model_accuracy_by_subject(model_name, num_examples):
|
73 |
try:
|
74 |
# Load the model and tokenizer
|
|
|
68 |
top_option = max(option_logits, key=lambda x: x[0])[1]
|
69 |
return top_option
|
70 |
|
71 |
+
@spaces.GPU(duration=120)
|
72 |
def evaluate_model_accuracy_by_subject(model_name, num_examples):
|
73 |
try:
|
74 |
# Load the model and tokenizer
|