Update README.md

README.md

@@ -1,14 +1,12 @@
 # SliceX AI™ ELM (Efficient Language Models)
 This repository contains code to run our ELM models.
 
-Models are located in the "models" folder. ELM models in this repository comes in three sizes (elm-1.0, elm-0.75 and elm-0.25) and supports the following use-cases.
-- news_classification
-- toxicity_detection
+Models are located in the "models" folder. ELM models in this repository comes in three sizes (elm-1.0, elm-0.75 and elm-0.25) and supports the following use-case.
 - news_content_generation
 
 ## Download ELM repo
 ```bash
-git clone [email protected]:slicexai/elm-0.25-v0.1
+git clone [email protected]:slicexai/elm-v0.1_news_content_generation
 sudo apt-get intall git-lfs 
 git lfs install
 ```
@@ -20,11 +18,11 @@ PATH=$PATH:/<absolute-path>/git-lfs-3.2.0/
 git lfs install
 ```
 
-## Download ELM task-specific model checkpoints
+## Download ELM model checkpoints
 ```bash
-cd elm-0.25-v0.1
-git lfs pull -I models/elm-0.25_news_classification/ckpt.pt
-git lfs pull -I models/elm-0.25_toxicity_detection/ckpt.pt
+cd elm-v0.1_news_content_generation
+git lfs pull -I models/elm-1.0_news_content_generation/ckpt.pt
+git lfs pull -I models/elm-0.75_news_content_generation/ckpt.pt
 git lfs pull -I models/elm-0.25_news_content_generation/ckpt.pt
 ```
 
@@ -36,13 +34,13 @@ pip install -r requirements.txt
 ## How to use - Run ELM on a sample task (e.g., news classification)
 ```bash
 python run.py <elm-model-directory>
-E.g. python run.py models/elm-0.25_news_classification
+E.g. python run.py models/elm-0.75_news_content_generation
 ``` 
-Prompts for the specific tasks can be found in the corresponding checkpoint directory. See an example below in the form of `models/elm-0.25_news_classification/example_prompts.json`.
+Prompts for the specific tasks can be found in the corresponding checkpoint directory. See an example below in the form of `models/elm-0.75_news_content_generation/example_prompts.json`.
 ```json
 {
-    "inputs": ["GM May Close Plant in Europe  DETROIT (Reuters) - General Motors Corp. &lt;A HREF=\"http://www.investor.reuters.com/FullQuote.aspx?ticker=GM.N target=/stocks/quickinfo/fullquote\"&gt;GM.N&lt;/A&gt; will likely  cut some jobs in Europe and may close a plant there as part of  a restructuring plan under development to try to return the  region to profitability, the U.S. automaker said on Wednesday."],
-    "template": "[INST]Below is a news article. Please classify it under one of the following classes (World, Business, Sports, Sci/Tech). Please format your response as a JSON payload.\n\n### Article: {input}\n\n### JSON Response:[/INST]"
+    "inputs": ["Scientists Invent 'Invisible' Metamaterial With Bonus Reflect Mode"],
+    "template": "[INST]The following headline is the headline of a news report. Please write the content of the news passage based on only this headline.\n\nHeadline: {input} \n\nContent:[/INST]"
 }
 ```
 
@@ -50,7 +48,7 @@ Running the above command returns the following response
 
 ```json
 {
-    "prompt": "[INST]Below is a news article. Please classify it under one of the following classes (World, Business, Sports, Sci/Tech). Please format your response as a JSON payload.\n\n### Article: GM May Close Plant in Europe  DETROIT (Reuters) - General Motors Corp. &lt;A HREF=\"http://www.investor.reuters.com/FullQuote.aspx?ticker=GM.N target=/stocks/quickinfo/fullquote\"&gt;GM.N&lt;/A&gt; will likely  cut some jobs in Europe and may close a plant there as part of  a restructuring plan under development to try to return the  region to profitability, the U.S. automaker said on Wednesday.\n\n### JSON Response:[/INST]",
-    "response": "{'text_label': 'Business'}"
+    "prompt": "[INST]The following headline is the headline of a news report. Please write the content of the news passage based on only this headline.\n\nHeadline: Scientists Invent 'Invisible' Metamaterial With Bonus Reflect Mode \n\nContent:[/INST]",
+    "response": "A team of scientists have created an invisible material that can make objects disappear. It is made of a special material that creates a layer of nanoscale dots that allow light to enter from the material, directing it to a layer of gas that allows light to enter from the material. The material is able to levitate and roll off its surface without leaving the material. This technology could have many future applications in battery technology, microelectronics, and more. A video demonstrating the material is available in the article."
 }
 ```