aboutsummaryrefslogtreecommitdiff
path: root/models/tensorspeare
diff options
context:
space:
mode:
Diffstat (limited to 'models/tensorspeare')
-rw-r--r--models/tensorspeare/index.html12
-rw-r--r--models/tensorspeare/index.md26
-rw-r--r--models/tensorspeare/tensorspeare.js1
3 files changed, 26 insertions, 13 deletions
diff --git a/models/tensorspeare/index.html b/models/tensorspeare/index.html
deleted file mode 100644
index 89946a3..0000000
--- a/models/tensorspeare/index.html
+++ /dev/null
@@ -1,12 +0,0 @@
----
-layout: default
-title: Models
----
-
-<script src="/assets/js/tf.min.js"></script>
-<script src="tensorspeare.js"></script>
-
-<div style="display: flex; flex-direction: column; justify-content: center; align-items: center; width: 100vw; max-height: 100vh; gap: 10px; padding-top: 10px">
- <button id="mybutton" class="button" style="color: #fff; background-color: #6a6980; border: none; padding: 15px 32px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px;" onclick="runModel();">Generate</button>
- <textarea readonly id="generated_text" rows="10" style="width: 55%; min-width: 400px; background-color: #1f1d29; resize:none; color: #FFFFFF; outline: none"></textarea>
-</div>
diff --git a/models/tensorspeare/index.md b/models/tensorspeare/index.md
new file mode 100644
index 0000000..2f7f881
--- /dev/null
+++ b/models/tensorspeare/index.md
@@ -0,0 +1,26 @@
+---
+title: TensorSpeare
+layout: model
+---
+### TensorSpeare
+
+A TensorFlow model that tries to imitate Shakespeare.
+
+### Layers
+
+1. Embedding
+
+2. Gated Recurrent Unit
+
+3. Dense
+
+### Try It
+
+<script src="/assets/js/tf.min.js"></script>
+<script src="tensorspeare.js"></script>
+
+<div class="flex flex-col justify-left items-left">
+<textarea readonly id="generated_text" rows="10" style="min-width: 400px; background-color: #1f1d29; resize:none; color: #FFFFFF; outline: none"></textarea>
+&nbsp;
+<button id="mybutton" class="button" style="color: #fff; background-color: #6a6980; border: none; padding: 15px 32px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px;" onclick="runModel()">Generate</button>
+</div>
diff --git a/models/tensorspeare/tensorspeare.js b/models/tensorspeare/tensorspeare.js
index e14fd08..f87679e 100644
--- a/models/tensorspeare/tensorspeare.js
+++ b/models/tensorspeare/tensorspeare.js
@@ -30,5 +30,4 @@ async function runModel(num_generate) {
idx2char = ['\n', ' ', '!', '$', '&', "'", ',', '-', '.', '3', ':', ';', '?', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']
const model = await tf.loadLayersModel('/models/tensorspeare/jsmodel/model.json');
generate_text(model, idx2char, char2idx, btn, generated_text, num_generate=num_generate);
- btn.textContent = "Generate"
}