dbleek commited on
Commit
fba0675
·
1 Parent(s): f6cbd2d

update model path

Browse files
Files changed (1) hide show
  1. 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("/workspaces/cs-gy-6613-project/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
 
 
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