Spaces:

ilyshi
/

shad_ml2_2

Runtime error

ilyshi commited on Mar 24, 2022

Commit

e0b0602

•

1 Parent(s): 33ccd55

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ def make_prediction(text):
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
-model_name = "train_model2"
 model_path = model_name + '.zip'
 model.load_state_dict(
     torch.load(

 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
+model_name = "trained_model2"
 model_path = model_name + '.zip'
 model.load_state_dict(
     torch.load(