marcelbinz commited on
Commit
49ac4d3
·
verified ·
1 Parent(s): 69649ce

Delete app.py.txt

Browse files
Files changed (1) hide show
  1. app.py.txt +0 -17
app.py.txt DELETED
@@ -1,17 +0,0 @@
1
- import gradio as gr, transformers
2
-
3
- pipe = transformers.pipeline("text-generation",
4
- model="marcelbinz/Llama-3.1-Minitaur-8B")
5
-
6
- def infer(prompt, max_tokens):
7
- return pipe(prompt, max_new_tokens=int(max_tokens))[0]["generated_text"]
8
-
9
- demo = gr.Interface(
10
- fn=infer,
11
- inputs=[gr.Textbox(label="Prompt"),
12
- gr.Slider(5,128,32,step=1,label="Max tokens")],
13
- outputs="text",
14
- title="Minitaur",
15
- description="Just type and hit *Run*"
16
- )
17
- demo.launch()