Spaces:
Running
Running
Update index.html
Browse files- 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()
|