Hugging Face's logo

Spaces:
briaai
/
BRIA-2.2-ControlNet-Canny
Running on Zero

App Files Community
1
dvir-bria commited on
Commit
eb694be
·
verified ·
1 Parent(s): fac36c2

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -67,6 +67,7 @@ class Model:
67
  self.base_model_id = base_model_id
68
  self.task_name = task_name
69
  print(f'Loaded {base_model_id}...')
 
70
  return pipe
71
 
72
  def set_base_model(self, base_model_id: str) -> str:
 
67
  self.base_model_id = base_model_id
68
  self.task_name = task_name
69
  print(f'Loaded {base_model_id}...')
70
+ print(f'Loaded {model_id}...')
71
  return pipe
72
 
73
  def set_base_model(self, base_model_id: str) -> str: