File size: 348 Bytes
40fb01d b748808 1160c53 b748808 1160c53 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import gradio as gr
# Load the model
model = gr.Interface.load("models/roneneldan/TinyStories-Instruct-33M")
# Create the interface
interface = gr.Interface(
fn=model.predict,
inputs=gr.Textbox(prompt="Type a prompt here..."),
outputs=gr.Textbox(),
live=True,
capture_session=True
)
# Launch the interface
interface.launch()
|