Update app.py

app.py

@@ -7,9 +7,13 @@ from tensorflow import keras
 from tensorflow.keras import layers
 app = FastAPI()
 
+'''
 model = from_pretrained_keras("mat27/medmnistPrueba")
 model.compile(loss="categorical_crossentropy", optimizer="adam", metrics=["accuracy"])
+'''
 
+pipe = pipeline(model="mat27/medmnistPrueba")
 @app.get("/generate")
 def generate(example: np.array):
-    print(model.predict(example))
+    output = pipe(example)
+    print(output)