Spaces:
Running
Running
| <html> | |
| <head> | |
| <link rel="stylesheet" href="https://fonts.googleapis.com/css?family=Roboto&display=swap" > | |
| <style> | |
| body { | |
| font-family: 'Roboto', sans-serif; | |
| font-size: 16px; | |
| } | |
| .logo { | |
| height: 1em; | |
| vertical-align: middle; | |
| margin-bottom: 0.1em; | |
| } | |
| </style> | |
| <script type="module" crossorigin src="https://cdn.jsdelivr.net/npm/@gradio/lite@0.4.2/dist/lite.js"></script> | |
| <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@gradio/lite@0.4.2/dist/lite.css" /> | |
| <link rel="stylesheet" href="https://gradio-hello-world.hf.space/theme.css"> | |
| </head> | |
| <body> | |
| <gradio-lite> | |
| <gradio-requirements> | |
| transformers_js_py | |
| </gradio-requirements> | |
| <gradio-file name="app.py" entrypoint> | |
| from transformers_js import import_transformers_js | |
| import gradio as gr | |
| transformers = await import_transformers_js() | |
| pipeline = transformers.pipeline | |
| pipe = await pipeline('text-generation', 'Xenova/LaMini-Cerebras-590M') | |
| async def generate(text): | |
| return await pipe(text, {"temperature": 2.0, "max_new_tokens": 50, "repetition_penalty": 1.5}) | |
| demo = gr.Interface( | |
| generate, | |
| gr.Textbox(label="Initial Prompt"), | |
| gr.JSON(label="Generated Text"), | |
| examples=["I was walking in a nice neighborhood the other day ", "I'm an evil penguin and I", "It wasn't a bad film but"], | |
| theme=gr.themes.Soft(), | |
| flag=False) | |
| demo.launch() | |
| </gradio-file> | |
| </gradio-lite> | |
| </body> | |
| </html> |