fix: model import
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import torch
|
|
8 |
from argparse import Namespace
|
9 |
from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
|
10 |
from io import BytesIO
|
11 |
-
from model import get_model
|
12 |
from src.utils.output_utils import prepare_output
|
13 |
from torchvision import transforms
|
14 |
|
|
|
8 |
from argparse import Namespace
|
9 |
from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
|
10 |
from io import BytesIO
|
11 |
+
from src.model import get_model
|
12 |
from src.utils.output_utils import prepare_output
|
13 |
from torchvision import transforms
|
14 |
|