Spaces:
Paused
Paused
giorgio-caparvi
commited on
Commit
·
487c322
1
Parent(s):
05f601f
simulating parsing
Browse files- api/app.py +5 -6
api/app.py
CHANGED
@@ -4,6 +4,7 @@ import subprocess
|
|
4 |
import os
|
5 |
import io
|
6 |
from model.src.utils.arg_parser import eval_parse_args # Nuovo import corretto
|
|
|
7 |
|
8 |
from model.src import eval
|
9 |
|
@@ -58,11 +59,12 @@ def generate_design():
|
|
58 |
subprocess.run(command, check=True)'''
|
59 |
try:
|
60 |
# Creiamo una lista di argomenti come quelli che passeresti via CLI
|
61 |
-
|
|
|
62 |
'--dataset_path', '/api/model/assets/data/vitonhd',
|
63 |
'--batch_size', '1',
|
64 |
'--mixed_precision', 'fp16',
|
65 |
-
'--output_dir', '/api/output',
|
66 |
'--save_name', 'generato_paired',
|
67 |
'--num_workers_test', '4',
|
68 |
'--sketch_cond_rate', '0.2',
|
@@ -70,12 +72,9 @@ def generate_design():
|
|
70 |
'--start_cond_rate', '0.0',
|
71 |
'--test_order', 'paired'
|
72 |
]
|
73 |
-
|
74 |
-
# Simuliamo l'arg_parser con la lista di argomenti
|
75 |
-
args = eval_parse_args(args_list)
|
76 |
|
77 |
# Esegui la funzione `main()` di eval.py passando gli argomenti
|
78 |
-
final_image = eval.main(
|
79 |
|
80 |
# Save the image to the current directory
|
81 |
save_path = os.path.join(os.getcwd(), 'generated_image.jpg')
|
|
|
4 |
import os
|
5 |
import io
|
6 |
from model.src.utils.arg_parser import eval_parse_args # Nuovo import corretto
|
7 |
+
import sys
|
8 |
|
9 |
from model.src import eval
|
10 |
|
|
|
59 |
subprocess.run(command, check=True)'''
|
60 |
try:
|
61 |
# Creiamo una lista di argomenti come quelli che passeresti via CLI
|
62 |
+
sys.argv = [
|
63 |
+
'eval.py',
|
64 |
'--dataset_path', '/api/model/assets/data/vitonhd',
|
65 |
'--batch_size', '1',
|
66 |
'--mixed_precision', 'fp16',
|
67 |
+
'--output_dir', '/api/output/generato_paired_paired/images',
|
68 |
'--save_name', 'generato_paired',
|
69 |
'--num_workers_test', '4',
|
70 |
'--sketch_cond_rate', '0.2',
|
|
|
72 |
'--start_cond_rate', '0.0',
|
73 |
'--test_order', 'paired'
|
74 |
]
|
|
|
|
|
|
|
75 |
|
76 |
# Esegui la funzione `main()` di eval.py passando gli argomenti
|
77 |
+
final_image = eval.main()
|
78 |
|
79 |
# Save the image to the current directory
|
80 |
save_path = os.path.join(os.getcwd(), 'generated_image.jpg')
|