akhaliq HF Staff commited on
Commit
26877ea
·
1 Parent(s): a630925

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ os.mkdir("outputs")
11
  os.system("gdown --id '1bjulr4yfazDmT-ZnFLZPLDyTPVf8gCCU'")
12
 
13
  def inference(truncation,seeds):
14
- os.system("python gen_images.py --outdir=./outputs/ --trunc="+str(truncation)+" --seeds="+str(int(seeds))+" --network=network-snapshot.pkl")
15
  seeds = int(seeds)
16
  image = Image.open(f"./outputs/seed{seeds:04d}.png")
17
  return image
 
11
  os.system("gdown --id '1bjulr4yfazDmT-ZnFLZPLDyTPVf8gCCU'")
12
 
13
  def inference(truncation,seeds):
14
+ os.system("python gen_images.py --outdir=./outputs/ --trunc="+str(truncation)+" --seeds="+str(int(seeds))+" --network=https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/pokemon.pkl")
15
  seeds = int(seeds)
16
  image = Image.open(f"./outputs/seed{seeds:04d}.png")
17
  return image