Spaces:

WRX020510
/

CustomModel_yelp

Sleeping

WRX020510 commited on Mar 15, 2025

Commit

c25ac28

verified ·

1 Parent(s): 34cf25b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import numpy as np
 # Testing with the saved model
-model2 = AutoModelForSequenceClassification.from_pretrained("CustomModel_yelp",
                                                             num_labels=5)
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")

 import numpy as np
 # Testing with the saved model
+model2 = AutoModelForSequenceClassification.from_pretrained("WRX020510/CustomModel_yelp",
                                                             num_labels=5)
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")