Update app.py

app.py

@@ -20,23 +20,25 @@ def make(Flow_Bytes_per_s,Fwd_Packets_Length_Total,Init_Fwd_Win_Bytes,Packet_Len
 10:"UDP",
 11:"UDPLag",
 12:"WebDDoS"}
-    s=[]
-    s.append(l)
-    loaded_model = joblib.load('your_model.pkl')
-    predictions=loaded_model.predict(s)
-    # for i in range(len(predictions)):
-    #     print("Prediction",i+1,':',reverse_encoding[predictions[i]])
-    import csv
-    input_file = "predictions.csv"
-    new_row = l
-    new_row.append(reverse_encoding[predictions[0]])
-    with open(input_file, 'r', newline='') as infile:
-        reader = csv.reader(infile)
-        data = list(reader)
-    with open(input_file, 'a', newline='') as outfile:
-        writer = csv.writer(outfile)
-        writer.writerow(new_row)
-    return reverse_encoding[predictions[0]]
+    if l==[0]*15: return "Invalid Input"
+    else:
+        s=[]
+        s.append(l)
+        loaded_model = joblib.load('your_model.pkl')
+        predictions=loaded_model.predict(s)
+        # for i in range(len(predictions)):
+        #     print("Prediction",i+1,':',reverse_encoding[predictions[i]])
+        import csv
+        input_file = "predictions.csv"
+        new_row = l
+        new_row.append(reverse_encoding[predictions[0]])
+        with open(input_file, 'r', newline='') as infile:
+            reader = csv.reader(infile)
+            data = list(reader)
+        with open(input_file, 'a', newline='') as outfile:
+            writer = csv.writer(outfile)
+            writer.writerow(new_row)
+        return reverse_encoding[predictions[0]]
 headline ="""
 Department of Information Technology \n
 National Institute of Technology Karnataka \n
@@ -52,12 +54,6 @@ iface=gr.Interface(fn=make,
                   examples=[[9638.51,2088.0,-1,1130.1428,33.617596,0.0,0.0,0,-1,14,0.0,35.08846,0.0,96,0],
                            [21989.434,3552.0,257,442608.88,665.2886,1.0,1.0,0,386,20,6.0,877.0259,2.0,120,1],
                             [46730160.0,2944.0,-1,0.0,0.0,0.0,63.0,0,-1,-1062718972,0.0,0.0,63.0,-2125437944,0]
-                           ],
-                  css="""
-                  body{background-color: lightgreen}
-                  .input_text input{
-                  background-color: lightblue !important;
-                  }
-                  """)
+                           ])
 # iface.launch(share=True)
 iface.launch(share=True,debug=False,favicon_path="logo.png")