misterbrainley commited on
Commit
7ea3fea
1 Parent(s): 20bb009

renamed function

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ N = 16
14
  cvae = ConditionalVariationalAutoEncoder()
15
  cvae.load_state_dict(torch.load('cvae.pth'))
16
 
17
- def greet(class_, species_, gender_):
18
  c_i = classes.index(class_)
19
  s_i = species.index(species_)
20
  g_i = genders.index(gender_)
@@ -35,7 +35,7 @@ def greet(class_, species_, gender_):
35
  return Z
36
 
37
  demo = gr.Interface(
38
- fn=greet,
39
  inputs=[
40
  gr.Dropdown(classes),
41
  gr.Dropdown(species),
 
14
  cvae = ConditionalVariationalAutoEncoder()
15
  cvae.load_state_dict(torch.load('cvae.pth'))
16
 
17
+ def generate(class_, species_, gender_):
18
  c_i = classes.index(class_)
19
  s_i = species.index(species_)
20
  g_i = genders.index(gender_)
 
35
  return Z
36
 
37
  demo = gr.Interface(
38
+ fn=generate,
39
  inputs=[
40
  gr.Dropdown(classes),
41
  gr.Dropdown(species),