RohitGuptaAI commited on
Commit
2351c04
·
1 Parent(s): b01ccca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ def check_by_url(txt_url):
56
  normalized_content_with_style += normalized_content_without_style
57
  new_data = {"title": title, "content": normalized_content_with_style}
58
 
59
- model = DistilBertForSequenceClassification.from_pretrained(".")
60
  tokenizer = DistilBertTokenizer.from_pretrained(".")
61
 
62
  test_encodings = tokenizer.encode_plus(
 
56
  normalized_content_with_style += normalized_content_without_style
57
  new_data = {"title": title, "content": normalized_content_with_style}
58
 
59
+ model = DistilBertForSequenceClassification.from_pretrained("pytorch_model.bin")
60
  tokenizer = DistilBertTokenizer.from_pretrained(".")
61
 
62
  test_encodings = tokenizer.encode_plus(