Spaces:

FFZG-cleopatra
/

latvian-twitter-sentiment-classifier

Build error

thak123 commited on May 10, 2023

Commit

99ad178

1 Parent(s): c4543eb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ import gradio as gr
 device = config.device
 model = BERTBaseUncased()
 model.load_state_dict(torch.load(
-    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)))
 model.to(device)
 # T = tokenizer.TweetTokenizer(