import numpy as np
import time

from huggingface_hub import from_pretrained_keras
from tensorflow import keras
from tensorflow.keras import layers
app = FastAPI()

model = from_pretrained_keras("mat27/medmnsitPrueba")
model.compile(loss="categorical_crossentropy", optimizer="adam", metrics=["accuracy"])

@app.get("/generate")
def generate(x: np.array):
    print(model.predict(x))