Spaces:
Sleeping
Sleeping
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")
|