Spaces:

librarian-bots
/

new-datasets-in-machine-learning

Running

davanstrien HF staff commited on Oct 6, 2023

Commit

c9dab8a

•

1 Parent(s): 6c65e74

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ import os
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 CACHE_TIME = 60 * 60 * 12  # 12 hours
-MAX_RESULTS = 5_000
 @cached(cache=TTLCache(maxsize=10, ttl=CACHE_TIME))
@@ -33,8 +33,8 @@ def get_arxiv_result():
 def load_model():
-    return SetFitModel.from_pretrained("librarian-bots/is_new_dataset_teacher_model")
 def format_row_for_model(row):
     return f"TITLE: {row['title']} \n\nABSTRACT: {row['abstract']}"

 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 CACHE_TIME = 60 * 60 * 12  # 12 hours
+MAX_RESULTS = 1_000
 @cached(cache=TTLCache(maxsize=10, ttl=CACHE_TIME))
 def load_model():
+    return SetFitModel.from_pretrained("librarian-bots/is_new_dataset_t`eacher_model")
+`
 def format_row_for_model(row):
     return f"TITLE: {row['title']} \n\nABSTRACT: {row['abstract']}"