Spaces:
Running
Running
dbleek
commited on
Commit
·
fba0675
1
Parent(s):
f6cbd2d
update model path
Browse files- milestone-3.py +1 -1
milestone-3.py
CHANGED
@@ -22,7 +22,7 @@ dataset = filtered_dataset.shuffle(seed=42).select(range(20))
|
|
22 |
dataset = dataset.sort("patent_number")
|
23 |
|
24 |
# Create pipeline using model trainned on Colab
|
25 |
-
model = torch.load("
|
26 |
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
27 |
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
28 |
|
|
|
22 |
dataset = dataset.sort("patent_number")
|
23 |
|
24 |
# Create pipeline using model trainned on Colab
|
25 |
+
model = torch.load("patent_classifier.pt", map_location=torch.device('cpu'))
|
26 |
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
27 |
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
28 |
|