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 |
|