Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ os.system("wget https://github.com/nyx-ai/stylegan2-flax-tpu/releases/download/v
|
|
5 |
os.mkdir("generated_images")
|
6 |
|
7 |
def inference(seeds,truncation_psi):
|
8 |
-
os.system("python generate_images.py --checkpoint cookie-256.pkl --seeds"+
|
9 |
-
return "generated_images/"+seeds+".png"
|
10 |
|
11 |
gr.Interface(inference,["number",gr.Slider(step=0.1)],"image").launch()
|
|
|
5 |
os.mkdir("generated_images")
|
6 |
|
7 |
def inference(seeds,truncation_psi):
|
8 |
+
os.system("python generate_images.py --checkpoint cookie-256.pkl --seeds"+str(seeds)+ "--truncation_psi "+str(truncation_psi)+" --out_path generated_images/")
|
9 |
+
return "generated_images/"+str(seeds)+".png"
|
10 |
|
11 |
gr.Interface(inference,["number",gr.Slider(step=0.1)],"image").launch()
|