Upload README.md

README.md

@@ -1,15 +1,38 @@
 ---
 library_name: peft
 base_model: mistralai/Mistral-7B-v0.1
+datasets:
+- JulesBelveze/tldr_news
 ---
 
 # Model Card for Model ID
 
-Trained with [Ludwig.ai](https://ludwig.ai) and [Prdibase](https://predibase.com)!
+Trained with [Ludwig.ai](https://ludwig.ai) and [Predibase](https://predibase.com)!
 
 Given a passage from a news report generates a headline.
 
-Trained on: https://huggingface.co/datasets/JulesBelveze/tldr_news
+Try it in [LoRAX](https://github.com/predibase/lorax):
+
+```python
+from lorax import Client
+
+client = Client("http://<your_endpoint>")
+
+content = "<your news story>"
+
+prompt = f"""
+The following passage is content from a news report. Please summarize this
+passage in one sentence or less.
+
+Passage: {content}
+
+Summary:
+"""
+
+adapter_id = "tgaddair/mistral-7b-tldrnews-headlines-lora-r8"
+resp = client.generate(prompt, max_new_tokens=64, adapter_id=adapter_id)
+print(resp.generated_text)
+```