Update README.md

README.md

@@ -1,11 +1,32 @@
-# Parks LLM - US Parks Assistant 🏞️
+---
+license: apache-2.0
+base_model: microsoft/DialoGPT-medium
+tags:
+- conversational
+- national-parks
+- travel
+- recreation
+- parks
+- outdoor
+- assistant
+- fine-tuned
+language:
+- en
+library_name: transformers
+pipeline_tag: text-generation
+model_type: gpt2
+datasets:
+- custom
+---
 
+# Parks LLM - US Parks Assistant 🏞️
 A conversational AI model fine-tuned to help with US national parks, state parks, and recreation areas.
 
 ## Quick Start
-
 Load the model:
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 
 tokenizer = AutoTokenizer.from_pretrained("ajc2195/parks-llm")
 model = AutoModelForCausalLM.from_pretrained("ajc2195/parks-llm")
@@ -25,19 +46,23 @@ def ask_parks_llm(question):
     
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response.split("Assistant:")[-1].strip()
+```
 
 Example usage:
+```python
 answer = ask_parks_llm("What should I pack for Yellowstone?")
 print(answer)
+```
 
 ## Pipeline Usage
-
+```python
 from transformers import pipeline
+
 parks_assistant = pipeline("text-generation", model="ajc2195/parks-llm")
 response = parks_assistant("Human: What's the best time to visit national parks?\nAssistant:")
+```
 
 ## Capabilities
-
 🏞️ Park Information: Details about national parks, state parks, and recreation areas  
 🎒 Trip Planning: Packing lists, preparation advice, and logistics  
 📅 Timing Advice: Best times to visit different parks  
@@ -48,7 +73,6 @@ response = parks_assistant("Human: What's the best time to visit national parks?
 🥾 Activities: Hiking, camping, and outdoor activity recommendations  
 
 ## Example Questions
-
 - What should I pack for a desert park visit?
 - How do I make campground reservations?
 - What are the best family-friendly national parks?
@@ -56,16 +80,14 @@ response = parks_assistant("Human: What's the best time to visit national parks?
 - What budget tips do you have for park visits?
 
 ## Model Details
-
 - Base Model: microsoft/DialoGPT-medium
 - Fine-tuned on: Comprehensive parks conversation dataset
 - Training Method: Manual PyTorch fine-tuning
 - Domain: US Parks and Recreation
 
 ## Limitations
-
 - Focuses primarily on US parks
 - General guidance - always verify with official park sources
 - Not a substitute for real-time park information or alerts
 
-Built with ❤️ for park enthusiasts and outdoor adventurers!
+Built with ❤️ for park enthusiasts and outdoor adventurers!