Update app.py

app.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI
 from transformers import pipeline
+import numpy as np
 import os
 os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
  
@@ -8,7 +9,10 @@ app = FastAPI()
  
 pipe = pipeline(model="mat27/medmnistPrueba")
 
-@app.get("/generate")
-def generate(){
-    
-}
+def generate(image: np.array):
+    """
+    Input: imagen
+    """
+    output = pipe(image)
+     
+    return {"output": output[0]}