KarthikAI commited on
Commit
6fb7a18
verified
1 Parent(s): 1cbd893

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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")