Spaces:
Runtime error
Runtime error
Commit
•
7ea3fea
1
Parent(s):
20bb009
renamed function
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ N = 16
|
|
14 |
cvae = ConditionalVariationalAutoEncoder()
|
15 |
cvae.load_state_dict(torch.load('cvae.pth'))
|
16 |
|
17 |
-
def
|
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=
|
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),
|