app.py

from fastapi import FastAPI
from transformers import pipeline
import numpy as np
import os
os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
 
# Create a new FastAPI app instance
app = FastAPI()
 
pipe = pipeline(model="mat27/medmnistPrueba")