Update index.html

index.html

@@ -21,12 +21,12 @@ transformers = await import_transformers_js()
 pipeline = transformers.pipeline
 pipe = await pipeline('zero-shot-image-classification')
 
-async def transcribe(text):
-	data = await pipe(text)
+async def classify(image, classes):
+	data = await pipe(image, classes.split(","))
 	result = [{item['label']: round(item['score'], 2) for item in data}]
 	return result
 
-demo = gr.Interface(transcribe, [gr.Image(label="Input image"), gr.Textbox(label="Classes")], gr.Label())
+demo = gr.Interface(transcribe, [gr.Image(label="Input image", type="filepath"), gr.Textbox(label="Classes separated by commas")], gr.Label())
 demo.launch()
 </gradio-file>