Spaces:

FFZG-cleopatra
/

lv-sentiment-demo

Sleeping

thak123 commited on Dec 12, 2022

Commit

b2aa5d6

1 Parent(s): 55b119f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ T = tokenizer.TweetTokenizer(preserve_handles=True, preserve_hashes=True, preser
 device = config.device
 model = BERTBaseUncased()
-model.load_state_dict(torch.load(model_path, map_location=torch.device(device)))
 model.to(device)
 def preprocess(text):

 device = config.device
 model = BERTBaseUncased()
+model.load_state_dict(torch.load(config.MODEL_PATH, map_location=torch.device(device)))
 model.to(device)
 def preprocess(text):