Spaces:

chbsaikiran
/

SMOLLM2_105M

Sleeping

chbsaikiran commited on Feb 13

Commit

f643792

1 Parent(s): 9b3af55

fixed app.py bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,8 @@ device = "cpu"
 checkpoint_path = "model_bin.pth"
 checkpoint = torch.load(checkpoint_path, map_location=device)
-model.load_state_dict(checkpoint['model_state_dict'])
 model.to(device)
 model.eval()

 checkpoint_path = "model_bin.pth"
 checkpoint = torch.load(checkpoint_path, map_location=device)
+#model.load_state_dict(checkpoint['model_state_dict'])
+model.load_state_dict(torch.load(checkpoint_path, map_location=device))
 model.to(device)
 model.eval()