Spaces:

shivamjadhav
/

albert_latest_96

Sleeping

shivamjadhav commited on Nov 10, 2024

Commit

9716ed4

1 Parent(s): d31e714

updated the model labels and api code

Files changed (1) hide show

classifier/Albert_latest.py CHANGED Viewed

@@ -33,7 +33,7 @@ class Model:
         return predictions[0].tolist()[:2],Labels[np.argmax(predictions)]
 model_instance = None
-model_weights = "../assets/albert_sentiment_checkpoint_58.pt"
 def get_model():
     global model_instance

         return predictions[0].tolist()[:2],Labels[np.argmax(predictions)]
 model_instance = None
+model_weights = "assets/albert_sentiment_checkpoint_58.pt"
 def get_model():
     global model_instance