Spaces:

FFZG-cleopatra
/

latvian-twitter-sentiment-classifier

Build error

thak123 commited on May 11, 2023

Commit

db7d8dd

1 Parent(s): f1749a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,7 @@ import gradio as gr
 device = config.device
 model = BERTBaseUncased()
-model.load_state_dict(torch.load(
-    config.MODEL_PATH, map_location=torch.device(device)))
 model.to(device)
 # T = tokenizer.TweetTokenizer(

 device = config.device
 model = BERTBaseUncased()
+model.load_state_dict(torch.load(config.MODEL_PATH, map_location=torch.device(device)),strict=False)
 model.to(device)
 # T = tokenizer.TweetTokenizer(