Spaces:
Sleeping
Sleeping
| import os, threading | |
| import gradio as gr | |
| from transformers import pipeline | |
| lock = threading.Lock() | |
| pipe = pipeline("image-to-text", | |
| model="Salesforce/blip-image-captioning-base") | |
| def exec(input): | |
| with lock: | |
| out = pipe(input) | |
| return out[0]["generated_text"] | |
| demo = gr.Interface(exec, | |
| inputs=gr.Image(type="pil", | |
| value="https://raw.githubusercontent.com/bstraehle/ai-ml-dl/main/hugging-face/hugging-face/beach.jpg"), | |
| outputs=[gr.Textbox(label = "output", value=os.environ["OUTPUT"])], | |
| description=os.environ["DESCRIPTION"]) | |
| demo.launch(mcp_server=True) |