Unknowncoode
commited on
Commit
·
c3d2571
1
Parent(s):
bee7f60
Add model card
Browse files
README.md
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
---
|
| 2 |
library_name: transformers
|
| 3 |
tags: []
|
|
@@ -196,4 +197,36 @@ Carbon emissions can be estimated using the [Machine Learning Impact calculator]
|
|
| 196 |
|
| 197 |
## Model Card Contact
|
| 198 |
|
| 199 |
-
[More Information Needed]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
<<<<<<< HEAD
|
| 2 |
---
|
| 3 |
library_name: transformers
|
| 4 |
tags: []
|
|
|
|
| 197 |
|
| 198 |
## Model Card Contact
|
| 199 |
|
| 200 |
+
[More Information Needed]
|
| 201 |
+
=======
|
| 202 |
+
# My DialoGPT Model
|
| 203 |
+
|
| 204 |
+
This is a fine-tuned version of `microsoft/DialoGPT-small` on custom data about Dominica.
|
| 205 |
+
|
| 206 |
+
## Model Details
|
| 207 |
+
|
| 208 |
+
- **Model Name**: DialoGPT-small
|
| 209 |
+
- **Training Data**: Custom dataset about Dominica
|
| 210 |
+
- **Evaluation**: Achieved `eval_loss` of 12.85
|
| 211 |
+
|
| 212 |
+
## Usage
|
| 213 |
+
|
| 214 |
+
To use this model, you can load it as follows:
|
| 215 |
+
|
| 216 |
+
```python
|
| 217 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 218 |
+
|
| 219 |
+
# Load the model and tokenizer
|
| 220 |
+
model_name = "unknownCode/IslandBoyRepo"
|
| 221 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 222 |
+
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 223 |
+
|
| 224 |
+
# Generate a response
|
| 225 |
+
input_text = "What is the capital of Dominica?"
|
| 226 |
+
inputs = tokenizer(input_text, return_tensors="pt")
|
| 227 |
+
outputs = model.generate(**inputs)
|
| 228 |
+
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 229 |
+
|
| 230 |
+
print(response)
|
| 231 |
+
```
|
| 232 |
+
>>>>>>> 7f19bf1 (Add model card)
|