mat27 commited on
Commit
ff981ea
·
1 Parent(s): 0420c68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,5 +11,5 @@ model = from_pretrained_keras("mat27/medmnsitPrueba")
11
  model.compile(loss="categorical_crossentropy", optimizer="adam", metrics=["accuracy"])
12
 
13
  @app.get("/generate")
14
- def generate(x: np.array):
15
- print(model.predict(x))
 
11
  model.compile(loss="categorical_crossentropy", optimizer="adam", metrics=["accuracy"])
12
 
13
  @app.get("/generate")
14
+ def generate(example: np.array):
15
+ print(model.predict(example))