updated epochs to 10

app.py

@@ -51,7 +51,7 @@ def malware_detection_DL (results, malicious_traffic, benign_traffic):
         metrics = ['accuracy'])
         #model.fit(x_train, y_train, batch_size=32, epochs = 150, validation_data=(x_test, y_test))
         #does not output epochs and gives evalutaion of validation data and history of losses and accuracy
-    history = model.fit(x_train, y_train, batch_size=32, epochs = 50,verbose=0, validation_data=(x_test, y_test))
+    history = model.fit(x_train, y_train, batch_size=32, epochs = 10,verbose=0, validation_data=(x_test, y_test))
     _, accuracy = model.evaluate(x_train, y_train)
         #return history.history
     if results=="Accuracy":