Hugging Face's logo

Spaces:
dgoot
/
image-to-image
Runtime error

App Files Community
dgoot commited on
Commit
a024324
·
1 Parent(s): 58c6bee

Bug: load the right model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def load_pipeline(model):
35
 
36
 
37
  logger.debug(f"Loading pipeline: {dict(model=model)}")
38
- pipe = load_pipeline(DEFAULT_MODEL).to("cuda")
39
 
40
 
41
  @logger.catch(reraise=True)
 
35
 
36
 
37
  logger.debug(f"Loading pipeline: {dict(model=model)}")
38
+ pipe = load_pipeline(model).to("cuda")
39
 
40
 
41
  @logger.catch(reraise=True)