Update main.py

main.py

@@ -33,7 +33,7 @@ import numpy as np  # Add this import
 import pandas as pd
 from io import BytesIO
 # main.py
-
+import torch
 # Standard library imports
 import io
 import re