Omartificial-Intelligence-Space commited on
Commit
83fc769
·
verified ·
1 Parent(s): cafafe0

update submit

Browse files
Files changed (1) hide show
  1. 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=300)
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