Commit
2500fde
1 Parent(s): 3b12908

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ datasets = {
50
  "Positive_Negatives_Dataset": load_dataset("Omartificial-Intelligence-Space/re-ar-query-candidate" , token =HF_TOKEN )["train"].select(range(500))
51
  }
52
 
53
- @spaces.GPU
54
  def evaluate_model_with_insights(model_name):
55
  model = CrossEncoder(model_name, device=device)
56
  results = []
 
50
  "Positive_Negatives_Dataset": load_dataset("Omartificial-Intelligence-Space/re-ar-query-candidate" , token =HF_TOKEN )["train"].select(range(500))
51
  }
52
 
53
+ @spaces.GPU(duration=120)
54
  def evaluate_model_with_insights(model_name):
55
  model = CrossEncoder(model_name, device=device)
56
  results = []