graykode

(add) javascript support in google colab

...@@ -56,8 +56,15 @@ ...@@ -56,8 +56,15 @@
56 "#### Download model weights\n", 56 "#### Download model weights\n",
57 "\n", 57 "\n",
58 "Download the two weights of model from the google drive through the gdown module.\n", 58 "Download the two weights of model from the google drive through the gdown module.\n",
59 - "1. [Added model](https://drive.google.com/uc?id=1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4) : A model trained Code2NL on Python using pre-trained CodeBERT (Feng at al, 2020).\n", 59 + "1. Added model : A model trained Code2NL on Python using pre-trained CodeBERT (Feng at al, 2020).\n",
60 - "2. [Diff model](https://drive.google.com/uc?id=1--gcVVix92_Fp75A-mWH0pJS0ahlni5m) : A model retrained by initializing with the weight of model (1), adding embedding of the added and deleted parts(`patch_ids_embedding`) of the code." 60 + "2. Diff model : A model retrained by initializing with the weight of model (1), adding embedding of the added and deleted parts(`patch_ids_embedding`) of the code.\n",
61 + "\n",
62 + "Download pre-trained weight\n",
63 + "\n",
64 + "Language | Added | Diff\n",
65 + "--- | --- | ---\n",
66 + "python | 1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4 | 1--gcVVix92_Fp75A-mWH0pJS0ahlni5m\n",
67 + "javascript | 1-F68ymKxZ-htCzQ8_Y9iHexs2SJmP5Gc | 1-39rmu-3clwebNURMQGMt-oM4HsAkbsf"
61 ] 68 ]
62 }, 69 },
63 { 70 {
...@@ -66,9 +73,12 @@ ...@@ -66,9 +73,12 @@
66 "id": "P9-EBpxt0Dp0" 73 "id": "P9-EBpxt0Dp0"
67 }, 74 },
68 "source": [ 75 "source": [
76 + "ADD_MODEL='1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4'\n",
77 + "DIFF_MODEL='1--gcVVix92_Fp75A-mWH0pJS0ahlni5m'\n",
78 + "\n",
69 "!pip install gdown \\\n", 79 "!pip install gdown \\\n",
70 - " && gdown \"https://drive.google.com/uc?id=1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4\" -O weight/added/pytorch_model.bin \\\n", 80 + " && gdown \"https://drive.google.com/uc?id=$ADD_MODEL\" -O weight/added/pytorch_model.bin \\\n",
71 - " && gdown \"https://drive.google.com/uc?id=1--gcVVix92_Fp75A-mWH0pJS0ahlni5m\" -O weight/diff/pytorch_model.bin" 81 + " && gdown \"https://drive.google.com/uc?id=$DIFF_MODEL\" -O weight/diff/pytorch_model.bin"
72 ], 82 ],
73 "execution_count": null, 83 "execution_count": null,
74 "outputs": [] 84 "outputs": []
......