Update app.py

app.py

@@ -4,13 +4,14 @@ import time
 
 from tensorflow import keras
 from tensorflow.keras import layers
+app = FastAPI()
 
 
 # Model / data parameters
 num_classes = 9
 input_shape = (28, 28, 3)
 batch_size = 1000
-epochs = 
+epochs = 1
 
 # Define baseline model
 def baseline_model():
@@ -73,8 +74,6 @@ history = model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, vali
 fin = time.time()
 print(fin-inicio)
 
-# Evaluation of the model
-score = model.evaluate(x_test, y_test, verbose=0)
-
-print("Test loss:", score[0])
-print("Test accuracy:", score[1])
+@app.get("/generate")
+def generate(x: np.array):
+    print(model.predict(x))