Spaces:
Paused
Paused
giorgio-caparvi
commited on
Commit
·
91bafa0
1
Parent(s):
9abe70d
fixed import
Browse files- api/model/src/eval.py +6 -6
- 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}")
|