Spaces:

junhyun01
/

Onpremise_LLM_Normal_Detection

Sleeping

junhyun01 commited on Jan 10, 2024

Commit

d747df2

•

1 Parent(s): 8854d9d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def paragraph_leveling(text):
             return out
     classifier = MLP(target_size=3, input_size=768)
-    classifier.load_state_dict(torch.load('./trained_model/fine_tunning_classifier'))
     classifier.to('cpu')
     classifier.eval()

             return out
     classifier = MLP(target_size=3, input_size=768)
+    classifier.load_state_dict(torch.load('./trained_model/fine_tunning_classifier', map_location=torch.device('cpu')))
     classifier.to('cpu')
     classifier.eval()