playgr
commited on
Commit
·
bae2d47
1
Parent(s):
8dd41e2
first test
Browse files- Dockerfile +3 -1
- app.py +4 -0
Dockerfile
CHANGED
@@ -26,4 +26,6 @@ RUN pip install --no-cache-dir \
|
|
26 |
# ENV HF_TOKEN=tu_token_aquí
|
27 |
|
28 |
# Arranca directamente el script
|
29 |
-
CMD ["uvicorn", "app:app",
|
|
|
|
|
|
26 |
# ENV HF_TOKEN=tu_token_aquí
|
27 |
|
28 |
# Arranca directamente el script
|
29 |
+
CMD ["uvicorn", "app:app", \
|
30 |
+
"--host", "0.0.0.0", "--port", "7860", \
|
31 |
+
"--log-level", "debug", "--reload"]
|
app.py
CHANGED
@@ -18,6 +18,10 @@ app = FastAPI()
|
|
18 |
CLOTHING_LABELS = ["a piece of clothing", "shirt", "jacket", "pants", "dress", "skirt"]
|
19 |
CLIP_THRESHOLD = 0.25
|
20 |
|
|
|
|
|
|
|
|
|
21 |
def process_image(pil_img: Image.Image, prompt: str, neg_prompt: str, hf_repo: str = None):
|
22 |
# --- Configuración de dispositivo
|
23 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
18 |
CLOTHING_LABELS = ["a piece of clothing", "shirt", "jacket", "pants", "dress", "skirt"]
|
19 |
CLIP_THRESHOLD = 0.25
|
20 |
|
21 |
+
print(" Starting app.py…")
|
22 |
+
|
23 |
+
|
24 |
+
|
25 |
def process_image(pil_img: Image.Image, prompt: str, neg_prompt: str, hf_repo: str = None):
|
26 |
# --- Configuración de dispositivo
|
27 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|