Unknowncoode commited on
Commit
c3d2571
·
1 Parent(s): bee7f60

Add model card

Browse files
Files changed (1) hide show
  1. README.md +34 -1
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)