Update main.py

main.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, File, UploadFile, Form
 from fastapi.responses import StreamingResponse
 from fastapi.staticfiles import StaticFiles
-
+import torch
 import shutil
 import cv2
 import numpy as np