Update index.html

index.html

@@ -33,10 +33,10 @@ transformers = await import_transformers_js()
 pipeline = transformers.pipeline
 pipe = await pipeline('text-generation', 'Xenova/LaMini-Cerebras-590M')
 
-async def classify(text):
+async def generate(text):
 	return await pipe(text, {"temperature": 2.0, "max_new_tokens": 50, "repetition_penalty": 1.5})
 
-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"])
+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"])
 demo.launch()
 </gradio-file>