Spaces:
Sleeping
Sleeping
smhavens
commited on
Commit
·
dc0c8b8
1
Parent(s):
bfc9cbd
dataset name fix
Browse files
app.py
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
-
import spacy
|
| 3 |
import math
|
| 4 |
from datasets import load_dataset
|
| 5 |
from sentence_transformers import SentenceTransformer
|
|
@@ -40,7 +39,7 @@ def compute_metrics(eval_pred):
|
|
| 40 |
|
| 41 |
|
| 42 |
def training():
|
| 43 |
-
dataset_id = "
|
| 44 |
dataset = load_dataset(dataset_id)
|
| 45 |
# dataset = dataset["train"]
|
| 46 |
# tokenized_datasets = dataset.map(tokenize_function, batched=True)
|
|
|
|
| 1 |
import gradio as gr
|
|
|
|
| 2 |
import math
|
| 3 |
from datasets import load_dataset
|
| 4 |
from sentence_transformers import SentenceTransformer
|
|
|
|
| 39 |
|
| 40 |
|
| 41 |
def training():
|
| 42 |
+
dataset_id = "ag_news"
|
| 43 |
dataset = load_dataset(dataset_id)
|
| 44 |
# dataset = dataset["train"]
|
| 45 |
# tokenized_datasets = dataset.map(tokenize_function, batched=True)
|