Hugging Face's logo

Spaces:
clemsou
/
pokemon_generator
Running

App Files Community
clemsou commited on
Commit
1803b4e
•
1 Parent(s): 916546a

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_3000v1-5.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_1300v1-5.h5",map_location=torch.device('cpu')))
46
 
47
 
48
  def predict(seed, pokemon_count):