diff --git a/main.py b/main.py
index 4ef8ab0b451b8beb82d19fdeed48a6f71f99da7e..a8f8fe29db609c92221fda33f5ed7c96fcf7862d 100644
--- a/main.py
+++ b/main.py
@@ -3,7 +3,8 @@ import platform
 import socket
 import psutil
 import streamlit
-
+import numpy as np
+import matplotlib.pyplot as plt
 def main():
     parser = argparse.ArgumentParser(description='Description of your program')
     parser.add_argument('-i', '--input', help='Input file path', required=False)