vision_endpoint.py

from fastapi import FastAPI, File
from transformers import pipeline

app = FastAPI()
vision_model = pipeline("image-analysis", model="tmmdev/chart-vision-analyzer")

@app.post("/analyze")
async def analyze_chart(image: File):
    analysis = vision_model(image)
    return {"analysis": analysis}