Spaces:
Running
on
Zero
Running
on
Zero
Omartificial-Intelligence-Space
commited on
Commit
•
2500fde
1
Parent(s):
3b12908
update app.py
Browse files
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 = []
|