Spaces:

ybelkada
/

dlai-test-2

Sleeping

dlai-test-2 / app.py

ybelkada HF staff

Update app.py

cca1333 verified 4 months ago

No virus

303 Bytes

	import gradio as gr
	from transformers import pipeline

	pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")

	def launch(input):
	out = pipe(input)
	return out[0]['generated_text']

	iface = gr.Interface(launch, inputs=gr.Image(type='pil'), outputs="text")
	iface.launch()