Matanew1 commited on
Commit
71abaad
1 Parent(s): 091b1d9
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,9 +12,10 @@ app.add_middleware(
12
  allow_methods=["*"], # Allows all methods
13
  allow_headers=["*"], # Allows all headers
14
  )
 
15
 
16
  # Initialize the VQA pipeline
17
- vqa_pipe = pipeline("visual-question-answering", model="Salesforce/blip-vqa-capfilt-large", max_new_tokens=20)
18
 
19
 
20
  @app.post('/answer_question')
 
12
  allow_methods=["*"], # Allows all methods
13
  allow_headers=["*"], # Allows all headers
14
  )
15
+ vqa_pipe = pipeline("visual-question-answering", model="Salesforce/blip-vqa-capfilt-large", tokenizer="Salesforce/blip-vqa-capfilt-large")
16
 
17
  # Initialize the VQA pipeline
18
+ # vqa_pipe = pipeline("visual-question-answering", model="Salesforce/blip-vqa-capfilt-large", max_new_tokens=20)
19
 
20
 
21
  @app.post('/answer_question')