Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import io # 1. For byte stream handling (file uploads)
|
2 |
from fastapi import FastAPI, File, UploadFile, Form # 2. FastAPI imports for API endpoints and file handling
|
3 |
from fastapi.responses import JSONResponse # 3. Used to return errors as JSON
|
4 |
-
from transformers import BlipProcessor, BlipForConditionalGeneration # 4. BLIP for image captioning
|
5 |
from PIL import Image # 5. Pillow for image processing
|
6 |
import openai # 6. OpenAI library for DALL路E API calls
|
7 |
import os # 7. OS for environment variables
|
@@ -12,8 +12,8 @@ from face_to_prompt import extract_face_prompt
|
|
12 |
app = FastAPI()
|
13 |
|
14 |
# 9. Load BLIP processor and model at startup to avoid reloading on every request
|
15 |
-
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
16 |
-
model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
|
17 |
|
18 |
# 10. Get the OpenAI API key from environment variable
|
19 |
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
|
|
|
1 |
import io # 1. For byte stream handling (file uploads)
|
2 |
from fastapi import FastAPI, File, UploadFile, Form # 2. FastAPI imports for API endpoints and file handling
|
3 |
from fastapi.responses import JSONResponse # 3. Used to return errors as JSON
|
4 |
+
# from transformers import BlipProcessor, BlipForConditionalGeneration # 4. BLIP for image captioning
|
5 |
from PIL import Image # 5. Pillow for image processing
|
6 |
import openai # 6. OpenAI library for DALL路E API calls
|
7 |
import os # 7. OS for environment variables
|
|
|
12 |
app = FastAPI()
|
13 |
|
14 |
# 9. Load BLIP processor and model at startup to avoid reloading on every request
|
15 |
+
# processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
16 |
+
# model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
|
17 |
|
18 |
# 10. Get the OpenAI API key from environment variable
|
19 |
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
|