Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import BlipProcessor, BlipForConditionalGeneration | |
| processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base") | |
| model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base") | |
| def execute(image): | |
| model_inputs = processor(image, return_tensors="pt") | |
| output = processor.decode(model.generate(**model_inputs)[0], skip_special_tokens=True) | |
| return output | |
| iface = gr.Interface(fn=execute, inputs="image", outputs="textbox") | |
| iface.launch() |