Refactor project structure: move modules to 'src' directory and update import paths; add new image processing and caption generation functionalities.
Browse files- app.py +3 -3
- dockerfile +1 -1
- Color_extraction.py β src/Color_extraction.py +0 -0
- Generate_caption.py β src/Generate_caption.py +0 -0
- Generate_productName_description.py β src/Generate_productName_description.py +0 -0
- Generating_prompt.py β src/Generating_prompt.py +0 -0
- test.py β src/test.py +0 -0
- train.py β src/train.py +0 -0
app.py
CHANGED
|
@@ -8,9 +8,9 @@ import asyncio
|
|
| 8 |
import os
|
| 9 |
from concurrent.futures import ThreadPoolExecutor
|
| 10 |
from dotenv import load_dotenv
|
| 11 |
-
from Generate_caption import load_model_from_path, tokenizer_load
|
| 12 |
-
from Color_extraction import extract_colors
|
| 13 |
-
from Generate_productName_description import generate_product_name, generate_description, clean_response
|
| 14 |
from huggingface_hub import hf_hub_download
|
| 15 |
import tempfile
|
| 16 |
|
|
|
|
| 8 |
import os
|
| 9 |
from concurrent.futures import ThreadPoolExecutor
|
| 10 |
from dotenv import load_dotenv
|
| 11 |
+
from src.Generate_caption import load_model_from_path, tokenizer_load
|
| 12 |
+
from src.Color_extraction import extract_colors
|
| 13 |
+
from src.Generate_productName_description import generate_product_name, generate_description, clean_response
|
| 14 |
from huggingface_hub import hf_hub_download
|
| 15 |
import tempfile
|
| 16 |
|
dockerfile
CHANGED
|
@@ -13,4 +13,4 @@ COPY --chown=user ./requirements.txt requirements.txt
|
|
| 13 |
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
| 14 |
|
| 15 |
COPY --chown=user . /app
|
| 16 |
-
CMD ["uvicorn", "app
|
|
|
|
| 13 |
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
| 14 |
|
| 15 |
COPY --chown=user . /app
|
| 16 |
+
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
|
Color_extraction.py β src/Color_extraction.py
RENAMED
|
File without changes
|
Generate_caption.py β src/Generate_caption.py
RENAMED
|
File without changes
|
Generate_productName_description.py β src/Generate_productName_description.py
RENAMED
|
File without changes
|
Generating_prompt.py β src/Generating_prompt.py
RENAMED
|
File without changes
|
test.py β src/test.py
RENAMED
|
File without changes
|
train.py β src/train.py
RENAMED
|
File without changes
|