Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ model = Sequential([
|
|
| 22 |
Dense(1)
|
| 23 |
])
|
| 24 |
model.compile(optimizer="adam", loss="mse")
|
| 25 |
-
model.fit(x, y, epochs=
|
| 26 |
|
| 27 |
# Prediction function
|
| 28 |
def predict_next(n1, n2, n3):
|
|
|
|
| 22 |
Dense(1)
|
| 23 |
])
|
| 24 |
model.compile(optimizer="adam", loss="mse")
|
| 25 |
+
model.fit(x, y, epochs=1000, batch_size=512, verbose=0) # use 1 epoch for faster Hugging Face init
|
| 26 |
|
| 27 |
# Prediction function
|
| 28 |
def predict_next(n1, n2, n3):
|