Update main.py

main.py

@@ -1,7 +1,7 @@
 from flask import Flask
 import torch
 from torch import nn
-# import re
+import re
 # import numpy as np
 # import pandas as pd
 # from collections import OrderedDict