Spaces:

MasumBhuiyan
/

MyDockerAPI

Sleeping

MasumBhuiyan commited on 5 days ago

Commit

7074e72

verified ·

1 Parent(s): 299b551

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -67,7 +67,8 @@ class ABSA(nn.Module):
 model = ABSA(vocab_size=len(token_2_id.keys()), num_labels=3)
-model = model.load_state_dict(torch.load('model_weights.pth'))
 print("Model loaded successfully")
 app = FastAPI()

 model = ABSA(vocab_size=len(token_2_id.keys()), num_labels=3)
+model.load_state_dict(torch.load('model_weights.pth'))
+model.eval()
 print("Model loaded successfully")
 app = FastAPI()