rahulved commited on
Commit ·
becf3fa
1
Parent(s): a446101
Changed app.py
Browse files
app.py
CHANGED
|
@@ -19,7 +19,7 @@ with open(network_pkl_a, 'rb') as f:
|
|
| 19 |
G_a = pickle.load(f)['G_ema'].cpu() # torch.nn.Module
|
| 20 |
|
| 21 |
def gen_image(text):
|
| 22 |
-
if text=='show me':
|
| 23 |
return gen_image_helper(G_d)
|
| 24 |
else:
|
| 25 |
return gen_image_helper(G_a)
|
|
|
|
| 19 |
G_a = pickle.load(f)['G_ema'].cpu() # torch.nn.Module
|
| 20 |
|
| 21 |
def gen_image(text):
|
| 22 |
+
if text=='show me the thing':
|
| 23 |
return gen_image_helper(G_d)
|
| 24 |
else:
|
| 25 |
return gen_image_helper(G_a)
|