Spaces:

krishna195
/

just

Running

App Files Files Community

krishna195 commited on Apr 3

Commit

b732fba

verified ·

1 Parent(s): 2b1755b

Update index.html

Browse files

Files changed (1) hide show

index.html +25 -25

index.html CHANGED Viewed

@@ -3,52 +3,52 @@
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Text Generation with DistilGPT-2</title>
-</head>
-<body>
-    <h2>Text Generation using DistilGPT-2</h2>
-    <p id="status">⏳ Loading model, please wait...</p>
-    <textarea id="inputText" rows="3" cols="50" placeholder="Enter your text..."></textarea>
-    <button id="generateBtn" disabled>Generate Text</button>
-    <p><strong>Generated Output:</strong></p>
-    <p id="output"></p>
     <script type="module">
-        import { pipeline } from "@xenova/transformers";
         let generator;
-        async function initializeModel() {
-            document.getElementById("status").innerText = "⏳ Loading model, please wait...";
             try {
                 generator = await pipeline("text-generation", "Xenova/distilgpt2");
-                document.getElementById("status").innerText = "✅ Model loaded! Ready to generate text.";
                 document.getElementById("generateBtn").disabled = false;
             } catch (error) {
-                document.getElementById("status").innerText = "❌ Failed to load model.";
-                console.error("Model loading error:", error);
             }
         }
         async function generateText() {
-            const inputText = document.getElementById("inputText").value;
-            if (!inputText.trim()) {
                 alert("Please enter some text!");
                 return;
             }
             document.getElementById("output").innerText = "⏳ Generating...";
             try {
-                const output = await generator(inputText, { max_new_tokens: 64, do_sample: true });
                 document.getElementById("output").innerText = output[0].generated_text;
             } catch (error) {
-                document.getElementById("output").innerText = "❌ Error in generation.";
-                console.error("Text generation error:", error);
             }
         }
-        document.getElementById("generateBtn").addEventListener("click", generateText);
-        // Initialize model on page load
-        initializeModel();
     </script>
 </body>
 </html>

 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Simple Text Generation</title>
     <script type="module">
+        import { pipeline } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers@2.5.4';
         let generator;
+        // Load the model once
+        async function loadModel() {
+            document.getElementById("status").innerText = "⏳ Loading model...";
             try {
                 generator = await pipeline("text-generation", "Xenova/distilgpt2");
+                document.getElementById("status").innerText = "✅ Model loaded!";
                 document.getElementById("generateBtn").disabled = false;
             } catch (error) {
+                document.getElementById("status").innerText = "❌ Error loading model.";
+                console.error("Model load error:", error);
             }
         }
         async function generateText() {
+            const inputText = document.getElementById("inputText").value.trim();
+            if (!inputText) {
                 alert("Please enter some text!");
                 return;
             }
             document.getElementById("output").innerText = "⏳ Generating...";
             try {
+                const output = await generator(inputText, { max_new_tokens: 50, do_sample: true });
                 document.getElementById("output").innerText = output[0].generated_text;
             } catch (error) {
+                document.getElementById("output").innerText = "❌ Error generating text.";
+                console.error("Generation error:", error);
             }
         }
+        window.addEventListener("DOMContentLoaded", loadModel);
+        window.generateText = generateText;
     </script>
+</head>
+<body>
+    <h2>Simple Text Generation</h2>
+    <p id="status">⏳ Loading model...</p>
+    <textarea id="inputText" rows="3" cols="50" placeholder="Enter text..."></textarea>
+    <br><br>
+    <button id="generateBtn" onclick="generateText()" disabled>Generate</button>
+    <h3>Output:</h3>
+    <p id="output"></p>
 </body>
 </html>