Update README.md
Browse files
README.md
CHANGED
@@ -42,5 +42,6 @@ class BLIPNet(torch.nn.Module):
|
|
42 |
|
43 |
model = BLIPNet()
|
44 |
model.load_state_dict(torch.load("best_model_wts_path"), strict=False)
|
45 |
-
|
46 |
-
You need to input the sample in the same way as shown in the example provided at: BLIP-Math
|
|
|
|
42 |
|
43 |
model = BLIPNet()
|
44 |
model.load_state_dict(torch.load("best_model_wts_path"), strict=False)
|
45 |
+
|
46 |
+
You need to input the sample in the same way as shown in the example provided at: https://huggingface.co/uf-aice-lab/BLIP-Math
|
47 |
+
Then you can get the generated text and classification score simultaneously.
|