Spaces:

librarian-bots
/

new-datasets-in-machine-learning

Sleeping

davanstrien HF staff commited on Oct 6, 2023

Commit

e5cdc54

•

1 Parent(s): 4fd027c

typo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ def get_arxiv_result():
 def load_model():
-    return SetFitModel.from_pretrained("librarian-bots/is_new_dataset_t`eacher_model")
 def format_row_for_model(row):
@@ -120,7 +120,7 @@ The Space works by:
 - searching for papers on arXiv with the term `dataset` in the title + "machine learning" in the abstract
 - passing the abstract and title of the papers to a machine learning model that predicts if the paper is introducing a new dataset or not
-This Space is a WIP in progress. The model is not perfect, and the search query is not perfect. If you have  suggestions for how to improve this Space, please open a Discussion.\n\n"""
 with gr.Blocks() as demo:

 def load_model():
+    return SetFitModel.from_pretrained("librarian-bots/is_new_dataset_teacher_model")
 def format_row_for_model(row):
 - searching for papers on arXiv with the term `dataset` in the title + "machine learning" in the abstract
 - passing the abstract and title of the papers to a machine learning model that predicts if the paper is introducing a new dataset or not
+This Space is a work in progress. The model is not perfect, and the search query is not perfect. If you have  suggestions for how to improve this Space, please open a Discussion.\n\n"""
 with gr.Blocks() as demo: