giorgio-caparvi commited on
Commit
91bafa0
·
1 Parent(s): 9abe70d

fixed import

Browse files
Files changed (2) hide show
  1. api/model/src/eval.py +6 -6
  2. chiama/chiamata.py +16 -0
api/model/src/eval.py CHANGED
@@ -13,12 +13,12 @@ from transformers import CLIPTextModel, CLIPTokenizer
13
 
14
  # custom imports
15
  #from datasets.dresscode import DressCodeDataset
16
- from datasets.vitonhd import VitonHDDataset
17
- from mgd_pipelines.mgd_pipe import MGDPipe
18
- from mgd_pipelines.mgd_pipe_disentangled import MGDPipeDisentangled
19
- from utils.arg_parser import eval_parse_args
20
- from utils.image_from_pipe import generate_images_from_mgd_pipe
21
- from utils.set_seeds import set_seed
22
  from PIL import Image
23
 
24
  # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
 
13
 
14
  # custom imports
15
  #from datasets.dresscode import DressCodeDataset
16
+ from model.src.datasets.vitonhd import VitonHDDataset
17
+ from model.src.mgd_pipelines.mgd_pipe import MGDPipe
18
+ from model.src.mgd_pipelines.mgd_pipe_disentangled import MGDPipeDisentangled
19
+ from model.src.utils.arg_parser import eval_parse_args
20
+ from model.src.utils.image_from_pipe import generate_images_from_mgd_pipe
21
+ from model.src.utils.set_seeds import set_seed
22
  from PIL import Image
23
 
24
  # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
chiama/chiamata.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import requests
2
+
3
+ # URL dell'API Flask
4
+ url = "http://localhost:7860/generate-design" # Assicurati che l'API sia in esecuzione su questo URL
5
+
6
+ # Invia una richiesta POST all'API per generare l'immagine
7
+ response = requests.post(url)
8
+
9
+ # Verifica se la richiesta è andata a buon fine
10
+ if response.status_code == 200:
11
+ # Salva l'immagine ricevuta nella cartella corrente
12
+ with open("received_image.jpg", "wb") as f:
13
+ f.write(response.content)
14
+ print("Immagine salvata come received_image.jpg nella directory corrente.")
15
+ else:
16
+ print(f"Errore durante la richiesta. Status code: {response.status_code}, Dettagli: {response.text}")