danhergir commited on
Commit
e4038ea
1 Parent(s): f379395
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ st.sidebar.image('assets/logo.jpeg', width=200)
15
  st.sidebar.caption('Demo was just created.')
16
 
17
  # Loading model
18
- repo_id = 'ceyda/bfutterfly_cropped_uniq1K_512'
19
  gan_model = load_model(repo_id)
20
 
21
  # Generate 4 butterflies
 
15
  st.sidebar.caption('Demo was just created.')
16
 
17
  # Loading model
18
+ repo_id = 'ceyda/butterfly_cropped_uniq1K_512'
19
  gan_model = load_model(repo_id)
20
 
21
  # Generate 4 butterflies