Update app.py

app.py

@@ -4,20 +4,18 @@ import time
 # Load the model
 model = gr.Interface.load("models/roneneldan/TinyStories-Instruct-33M")
 
-# Define additional inputs and outputs, if applicable
-additional_input = gr.inputs.Textbox(label="Input Text", placeholder="Enter your story here...")
-additional_output = gr.outputs.Textbox(label="Generated Story")
+# 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
 
-# Function to simulate typing effect
-def typing_effect(text):
-    for char in text:
-        print(char, end='', flush=True)  # Print character without newline
-        time.sleep(0.05)  # Add a small delay between characters
-    print()  # Print newline after the text is complete
-
-# Launch the interface with added options and descriptions
-model.launch(inputs=additional_input, outputs=additional_output, title="Tiny Stories Generator", 
-             description="This model generates creative stories based on the input text provided.",
-             examples=[["Once upon a time..."]],
-             live=False,  # Disable live updates to control the typing effect
-             fn=typing_effect)  # Use the typing_effect function to display output
+# 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()