Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def make_prediction(text):
|
|
31 |
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
32 |
|
33 |
model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
|
34 |
-
model_name = "
|
35 |
model_path = model_name + '.zip'
|
36 |
model.load_state_dict(
|
37 |
torch.load(
|
|
|
31 |
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
32 |
|
33 |
model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
|
34 |
+
model_name = "trained_model2"
|
35 |
model_path = model_name + '.zip'
|
36 |
model.load_state_dict(
|
37 |
torch.load(
|