File size: 1,518 Bytes
9bcb716
 
 
ec8b5c1
 
 
 
 
 
 
 
 
 
 
 
 
9af2d39
 
a0ada6d
9bcb716
 
ec8b5c1
 
 
 
 
 
 
 
 
 
 
 
9af2d39
ec8b5c1
7dc629e
d6e3ae0
ec8b5c1
aa25822
 
 
 
 
 
 
ec8b5c1
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
<!DOCTYPE html>
<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/[email protected]/dist/lite.js"></script>
		<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@gradio/[email protected]/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>