Update README.md
Browse files
README.md
CHANGED
|
@@ -18,12 +18,32 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 18 |
|
| 19 |
# git-base-instagram-cap
|
| 20 |
|
| 21 |
-
This model is a fine-tuned version of [microsoft/git-base](https://huggingface.co/microsoft/git-base) on an
|
|
|
|
| 22 |
|
| 23 |
It achieves the following results on the evaluation set:
|
| 24 |
- Loss: 0.0093
|
| 25 |
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
| 28 |
### Framework versions
|
| 29 |
|
|
|
|
| 18 |
|
| 19 |
# git-base-instagram-cap
|
| 20 |
|
| 21 |
+
This model is a fine-tuned version of [microsoft/git-base](https://huggingface.co/microsoft/git-base) on an
|
| 22 |
+
[mrSoul7766/instagram_post_captions](https://huggingface.co/datasets/mrSoul7766/instagram_post_captions).
|
| 23 |
|
| 24 |
It achieves the following results on the evaluation set:
|
| 25 |
- Loss: 0.0093
|
| 26 |
|
| 27 |
|
| 28 |
+
### Usage
|
| 29 |
+
|
| 30 |
+
To use the fine-tuned chatbot model, you can leverage the capabilities provided by the Hugging Face Transformers library. Here's a basic example using Python:
|
| 31 |
+
|
| 32 |
+
```python
|
| 33 |
+
# Use a pipeline as a high-level helper
|
| 34 |
+
from transformers import pipeline
|
| 35 |
+
|
| 36 |
+
pipe = pipeline("image-to-text", model="mrSoul7766/git-base-instagram-cap")
|
| 37 |
+
|
| 38 |
+
# Generate caption
|
| 39 |
+
caption = pipe("/content/download (12).png",max_new_tokens =100)
|
| 40 |
+
|
| 41 |
+
# Print the generated answer
|
| 42 |
+
print(caption[0]['generated_text'])
|
| 43 |
+
```
|
| 44 |
+
```
|
| 45 |
+
i love my blonde character in kim kardashian hollywood! i'm playing now who's playing with me?
|
| 46 |
+
```
|
| 47 |
|
| 48 |
### Framework versions
|
| 49 |
|