akhaliq HF staff commited on
Commit
bd6ab07
1 Parent(s): 0e0db81

Update app.py

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