Update app.py

app.py

@@ -56,9 +56,9 @@ def malware_detection_DL (results, malicious_traffic, benign_traffic):
     model.compile(optimizer ='adam', 
         loss = 'binary_crossentropy',
         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))
+       
+        #does not output epochs and gives evalutaion of validation data and history of losses and accuracy, epochs lowered to 10 from 150 for speed in displaying plot
+    history = model.fit(x_train, y_train, batch_size=32, epochs = 30,verbose=0, validation_data=(x_test, y_test))
     _, accuracy = model.evaluate(x_train, y_train)
         #return history.history
     if results=="Accuracy":
@@ -91,4 +91,4 @@ iface = gr.Interface(
     
 )
 
-iface.launch()
+iface.launch(enable_queue =True)