Hugging Face's logo

Spaces:
ilar06
/
hfjsdemo_my-copy
Running

App Files Community
hfjsdemo_my-copy
File size: 6,526 Bytes 
d8d37b0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
 
<!DOCTYPE html>
<html lang="en">

<head>
    <meta charset="UTF-8">
    <title>Text Generation - Hugging Face Transformers.js</title>

    <script type="module">
        // Import the library
        import { pipeline } from 'https://cdn.jsdelivr.net/npm/@xenova/[email protected]';

        // Make it available globally
        window.pipeline = pipeline;
    </script>

    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet">

    <link rel="stylesheet" href="css/styles.css">
</head>

<body>
    <div class="container-main">
        <!-- Page Header -->
        <div class="header">
            <div class="header-logo">
                <img src="images/logo.png" alt="logo">
            </div>
            <div class="header-main-text">
                <h1>Hugging Face Transformers.js</h1>
            </div>
            <div class="header-sub-text">
                <h3>Free AI Models for JavaScript Web Development</h3>
            </div>
        </div>
        <hr> <!-- Separator -->

        <!-- Back to Home button -->
        <div class="row mt-5">
            <div class="col-md-12 text-center">
                <a href="index.html" class="btn btn-outline-secondary"
                    style="color: #3c650b; border-color: #3c650b;">Back to Main Page</a>
            </div>
        </div>

        <!-- Content -->
        <div class="container mt-5">
            <!-- Centered Titles -->
            <div class="text-center">
                <h2>Natural Language Processing</h2>
                <h4>Text Generation</h4>
            </div>

            <!-- Actual Content of this page -->
            <div id="text-generation-container" class="container mt-4">
                <h5>Text Generation with Xenova/distilgpt2 (Default Settings):</h5>
                <div class="d-flex align-items-center">
                    <label for="generationText" class="mb-0 text-nowrap" style="margin-right: 15px;">Enter Text to
                        Generate:</label>
                    <input type="text" class="form-control flex-grow-1" id="generationText"
                        value="I enjoy walking with my cute dog," placeholder="Enter text"
                        style="margin-right: 15px; margin-left: 15px;">
                    <button id="generateButton" class="btn btn-primary" onclick="generateText()">Generate</button>
                </div>
                <div class="mt-4">
                    <h4>Output:</h4>
                    <pre id="outputArea"></pre>
                </div>
            </div>

            <hr> <!-- Line Separator -->

            <div id="text-generation-container2" class="container mt-4">
                <h5>Text Generation with Xenova/distilgpt2 (Custom Settings):</h5>
                <div class="d-flex align-items-center">
                    <label for="generationText2" class="mb-0 text-nowrap" style="margin-right: 15px;">Enter Text to
                        Generate:</label>
                    <input type="text" class="form-control flex-grow-1" id="generationText2"
                        value="Once upon a time, there was" placeholder="Enter text"
                        style="margin-right: 15px; margin-left: 15px;">
                    <button id="generateButton2" class="btn btn-primary" onclick="generateText2()">Generate</button>
                </div>
                <div class="mt-4">
                    <h4>Output:</h4>
                    <pre id="outputArea2"></pre>
                </div>
            </div>

            <hr> <!-- Line Separator -->

            <div id="code-generation-container" class="container mt-4">
                <h5>Code Generation with Xenova/codegen-350M-mono:</h5>
                <div class="d-flex align-items-center">
                    <label for="generationCode" class="mb-0 text-nowrap" style="margin-right: 15px;">Enter code to
                        Generate:</label>
                    <input type="text" class="form-control flex-grow-1" id="generationCode"
                        value="def fib(n):" placeholder="Enter code"
                        style="margin-right: 15px; margin-left: 15px;">
                    <button id="generateCodeButton" class="btn btn-primary" onclick="generateCode()">Generate Code</button>
                </div>
                <div class="mt-4">
                    <h4>Output:</h4>
                    <pre id="outputAreaCode"></pre>
                </div>
            </div>

            <!-- Back to Home button -->
            <div class="row mt-5">
                <div class="col-md-12 text-center">
                    <a href="index.html" class="btn btn-outline-secondary"
                        style="color: #3c650b; border-color: #3c650b;">Back to Main Page</a>
                </div>
            </div>
        </div>
    </div>

    <script>

        let generator;
        let code_generator;

        // Initialize the sentiment analysis model
        async function initializeModel() {
            generator = await pipeline('text-generation', 'Xenova/distilgpt2');
            code_generator = await pipeline('text-generation', 'Xenova/codegen-350M-mono');

        }

        async function generateText() {
            const textFieldValue = document.getElementById("generationText").value.trim();

            const result = await generator(textFieldValue);

            document.getElementById("outputArea").innerText = JSON.stringify(result, null, 2);
        }

        async function generateText2() {
            const textFieldValue = document.getElementById("generationText2").value.trim();

            const result = await generator(textFieldValue, {
                temperature: 2,
                max_new_tokens: 20,
                repetition_penalty: 1.5,
                no_repeat_ngram_size: 2,
                num_beams: 2,
                num_return_sequences: 3,
            });

            document.getElementById("outputArea2").innerText = JSON.stringify(result, null, 2);
        }
        
        async function generateCode() {
            const textFieldValue = document.getElementById("generationCode").value.trim();

            const result = await code_generator(textFieldValue);

            document.getElementById("outputAreaCode").innerText = JSON.stringify(result, null, 2);
        }

        // Initialize the model after the DOM is completely loaded
        window.addEventListener("DOMContentLoaded", initializeModel);
    </script>
</body>

</html>