Spaces:

Ireneo
/

roneneldan-TinyStories-Instruct-33M

Sleeping

App Files Files Community

Ireneo commited on Aug 1, 2023

Commit

4f00760

•

1 Parent(s): 60b66fc

Update app.py

Browse files

Files changed (1) hide show

app.py +41 -15

app.py CHANGED Viewed

@@ -1,21 +1,47 @@
 import gradio as gr
-import time
 # Load the model
 model = gr.Interface.load("models/roneneldan/TinyStories-Instruct-33M")
-# Simulate the typing effect
-def simulate_typing(input_text):
-    typing_speed = 0.05  # Adjust the speed of typing
-    generated_text = model.predict(input_text)
-    for char in generated_text:
-        time.sleep(typing_speed)
-        # Update the text display area in the Android app with each character
-        # You'll need to implement this part in your Android app using appropriate methods.
-        # For example, you can update the TextView in Android with the new character.
-    return generated_text
-# Launch the interface
-gr.Interface(fn=simulate_typing, inputs=gr.inputs.Textbox(label="Input Text", placeholder="Enter your story here..."),
-             outputs=gr.outputs.Textbox(label="Generated Story"), title="Tiny Stories Generator",
-             description="This model generates creative stories based on the input text provided.").launch()

 import gradio as gr
 # Load the model
 model = gr.Interface.load("models/roneneldan/TinyStories-Instruct-33M")
+# HTML template with JavaScript to update the text dynamically
+html_code = """
+<!DOCTYPE html>
+<html>
+<head>
+    <title>Tiny Stories Generator</title>
+    <script>
+        function generateStory() {
+            var inputText = document.getElementById("input-text").value;
+            gr.Interface.static.predict(inputText).then(function(outputText) {
+                document.getElementById("output-text").innerHTML = "";
+                typeText(inputText + " ");
+                typeText(outputText);
+            });
+        }
+        function typeText(text) {
+            var index = 0;
+            var typingSpeed = 50; // Adjust the speed of typing
+            var timer = setInterval(function() {
+                if (index < text.length) {
+                    document.getElementById("output-text").innerHTML += text.charAt(index);
+                    index++;
+                } else {
+                    clearInterval(timer);
+                }
+            }, typingSpeed);
+        }
+    </script>
+</head>
+<body>
+    <div style="padding: 20px;">
+        <textarea id="input-text" placeholder="Enter your story here..." style="width: 100%; height: 200px;"></textarea>
+        <button onclick="generateStory()">Generate Story</button>
+        <div id="output-text" style="margin-top: 20px;"></div>
+    </div>
+</body>
+</html>
+"""
+# Launch the interface using the custom HTML code
+gr.Interface(fn=lambda inputs: inputs, inputs=gr.inputs.Textbox(), outputs=gr.outputs.HTML(html_code)).launch()