#!/usr/bin/env python # coding: utf-8 # In[2]: import gradio as gr import torch from torch import nn from torch.nn import functional as F from nano_gpt_inferencing import generate_paragraph HTML_TEMPLATE = """
Image 1
Image 2

Paragraph Auto Completion like Shakespeare

Generate dialogue using the intellignece from Shakespeare Dataset .

Model: GPT, Dataset: Tiny Shakespeare, Token limit: User input ,Input Text: User input.

""" with gr.Blocks(theme=gr.themes.Glass()) as interface: gr.HTML(value=HTML_TEMPLATE, show_label=False) with gr.Row(scale=1): inputs = [ gr.Textbox(label="Input Text Prompt"), gr.Textbox(label="Token Limit") ] outputs = gr.Textbox( label="Generated Paragraph" ) with gr.Column(scale=1): button = gr.Button("Generate") button.click(generate_paragraph, inputs=inputs, outputs=outputs) if __name__ == "__main__": interface.launch(enable_queue=True) # In[ ]: