Update app.py

app.py

@@ -1,7 +1,14 @@
 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}