Update main.py
Browse files
main.py
CHANGED
|
@@ -10,6 +10,9 @@ import tempfile
|
|
| 10 |
from mtcnn import MTCNN
|
| 11 |
from vtoonify_model import Model # Import VToonify model
|
| 12 |
|
|
|
|
|
|
|
|
|
|
| 13 |
app = FastAPI()
|
| 14 |
|
| 15 |
# Initialize the VToonify model and MTCNN detector
|
|
|
|
| 10 |
from mtcnn import MTCNN
|
| 11 |
from vtoonify_model import Model # Import VToonify model
|
| 12 |
|
| 13 |
+
import torch
|
| 14 |
+
|
| 15 |
+
|
| 16 |
app = FastAPI()
|
| 17 |
|
| 18 |
# Initialize the VToonify model and MTCNN detector
|