Robaa commited on
Commit
5d95877
Β·
1 Parent(s): d16a72e

Refactor project structure: move modules to 'src' directory and update import paths; add new image processing and caption generation functionalities.

Browse files
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.py", "--host", "0.0.0.0", "--port", "7860"]
 
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