Merge pull request #28 from jmisilo/develop

Develop
jmisilo · Oct 28, 2022 · adababc · adababc
2 parents 25438aa + 03a2deb
commit adababc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/examples/7757242158.jpg b/examples/7757242158.jpg
diff --git a/readme.md b/readme.md
@@ -18,6 +18,8 @@ The training process was carried out using the [Kaggle](https://www.kaggle.com/)
 
 ![Example2](./examples/3787801.jpg)
 
+![Example3](./examples/7757242158.jpg)
+
 As I said, the goal was to test the Model's ability to recognize the situation. In the next phase of the experiments, I will try to improve the Model process and parameters to achieve better captions with the same dataset.
 
 ### Usage

diff --git a/src/model/model.py b/src/model/model.py
@@ -197,7 +197,7 @@ def forward(self, img):
                 if last_token == self.td.tokenizer.eos_token_id:
                     break
 
-            decoded = self.td.tokenizer.decode(tokens[self.ep_len + 1:])
+            decoded = self.td.tokenizer.decode(tokens)
 
             return decoded, tokens