Hugging Face's logo

Spaces:
clemsou
/
pokemon_generator
Running

App Files Community
clemsou commited on
Commit
19a98a2
•
1 Parent(s): d8c0ad3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ class Generator_128(nn.Module):
42
 
43
 
44
  trained_gen = Generator_128(0)
45
- trained_gen.load_state_dict(torch.load("generator_epoch_1000v14.h5",map_location=torch.device('cpu')))
46
 
47
 
48
  def predict(seed, pokemon_count):
 
42
 
43
 
44
  trained_gen = Generator_128(0)
45
+ trained_gen.load_state_dict(torch.load("generator_epoch_1000.h5",map_location=torch.device('cpu')))
46
 
47
 
48
  def predict(seed, pokemon_count):