from fastapi import FastAPI
from transformers import pipeline
import numpy as np
 
# Create a new FastAPI app instance
app = FastAPI()
 
pipe = pipeline(model="mat27/medmnistPrueba")

@app.get("/generate")
def generate(imagen: np.array):
    output = pipe(imagen)
     
    return {"output": output}