giorgio-caparvi commited on
Commit
487c322
·
1 Parent(s): 05f601f

simulating parsing

Browse files
Files changed (1) hide show
  1. 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
- args_list = [
 
62
  '--dataset_path', '/api/model/assets/data/vitonhd',
63
  '--batch_size', '1',
64
  '--mixed_precision', 'fp16',
65
- '--output_dir', '/api/output', # Se serve ancora questo argomento
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(args)
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')