Matanew1 commited on
Commit
d711eec
1 Parent(s): 9bcbf71
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from fastapi import FastAPI, File, UploadFile, Form
2
  from fastapi.responses import JSONResponse
3
  from transformers import BlipProcessor, TFBlipForQuestionAnswering
@@ -18,10 +19,18 @@ app.add_middleware(
18
  processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-capfilt-large")
19
  model = TFBlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-capfilt-large")
20
 
 
 
 
 
 
 
 
 
21
 
22
  @app.post('/answer_question')
23
  async def answer_question(image: UploadFile = File(...), question: str = Form(...)):
24
- image_path = 'temp_image.jpg'
25
  with open(image_path, 'wb') as f:
26
  f.write(await image.read())
27
 
 
1
+ import os
2
  from fastapi import FastAPI, File, UploadFile, Form
3
  from fastapi.responses import JSONResponse
4
  from transformers import BlipProcessor, TFBlipForQuestionAnswering
 
19
  processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-capfilt-large")
20
  model = TFBlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-capfilt-large")
21
 
22
+ # Set directory path for storing images
23
+ image_directory = 'images'
24
+
25
+ # Create the directory if it doesn't exist
26
+ os.makedirs(image_directory, exist_ok=True)
27
+
28
+ # Set read and write permissions for the directory
29
+ os.chmod(image_directory, 0o777)
30
 
31
  @app.post('/answer_question')
32
  async def answer_question(image: UploadFile = File(...), question: str = Form(...)):
33
+ image_path = os.path.join(image_directory, 'temp_image.jpg')
34
  with open(image_path, 'wb') as f:
35
  f.write(await image.read())
36