Update main.py
Browse files
main.py
CHANGED
@@ -4,6 +4,7 @@ from fastapi.responses import FileResponse
|
|
4 |
from fastapi.staticfiles import StaticFiles
|
5 |
from fastapi import FastAPI, File, UploadFile, Form
|
6 |
from fastapi.responses import FileResponse
|
|
|
7 |
import shutil
|
8 |
import cv2
|
9 |
import numpy as np
|
|
|
4 |
from fastapi.staticfiles import StaticFiles
|
5 |
from fastapi import FastAPI, File, UploadFile, Form
|
6 |
from fastapi.responses import FileResponse
|
7 |
+
import torch
|
8 |
import shutil
|
9 |
import cv2
|
10 |
import numpy as np
|