reach-vb HF Staff commited on
Commit
d6e3ae0
·
1 Parent(s): a0ada6d

Update index.html

Browse files
Files changed (1) hide show
  1. index.html +1 -1
index.html CHANGED
@@ -34,7 +34,7 @@ pipeline = transformers.pipeline
34
  pipe = await pipeline('text-generation', 'Xenova/LaMini-Cerebras-590M')
35
 
36
  async def classify(text):
37
- return await pipe(text)
38
 
39
  demo = gr.Interface(classify, "textbox", "json", 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"])
40
  demo.launch()
 
34
  pipe = await pipeline('text-generation', 'Xenova/LaMini-Cerebras-590M')
35
 
36
  async def classify(text):
37
+ return await pipe(text, {"temperature": 2.0, "max_new_tokens": 50, "repetition_penalty": 1.5})
38
 
39
  demo = gr.Interface(classify, "textbox", "json", 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"])
40
  demo.launch()