#!/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 = """
"""
with gr.Blocks(theme=gr.themes.Glass(),css=".gradio-container {background: url('https://github.com/nkanungo/ERAS20/blob/main/images/bg_1.jpg?raw=true')}") 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():
button = gr.Button("Generate")
button.click(generate_paragraph, inputs=inputs, outputs=outputs)
if __name__ == "__main__":
interface.launch(enable_queue=True)
# In[ ]:
