Aguilar Elizondo commited on
Commit
ffbb1f9
·
1 Parent(s): 544d65f

Fix: Import DiffusionPipelineManager and PostProcessor correctly

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,9 +13,9 @@ import sys
13
  # Add backend to path
14
  sys.path.insert(0, str(Path(__file__).parent))
15
 
16
- from backend.services.diffusion_pipeline import DiffusionPipeline
17
  from backend.services.upscaler import ESRGANUpscaler
18
- from backend.services.post_processor import PostProcessor
19
  from backend.config import settings
20
 
21
  # Configure logging
@@ -33,7 +33,7 @@ def initialize_models():
33
 
34
  if diffusion_pipeline is None:
35
  logger.info("Loading Stable Diffusion pipeline...")
36
- diffusion_pipeline = DiffusionPipeline()
37
 
38
  if upscaler is None:
39
  logger.info("Loading upscaler...")
 
13
  # Add backend to path
14
  sys.path.insert(0, str(Path(__file__).parent))
15
 
16
+ from backend.services.diffusion_pipeline import DiffusionPipelineManager
17
  from backend.services.upscaler import ESRGANUpscaler
18
+ from backend.services.postprocess import PostProcessor
19
  from backend.config import settings
20
 
21
  # Configure logging
 
33
 
34
  if diffusion_pipeline is None:
35
  logger.info("Loading Stable Diffusion pipeline...")
36
+ diffusion_pipeline = DiffusionPipelineManager()
37
 
38
  if upscaler is None:
39
  logger.info("Loading upscaler...")