ikraamkb commited on
Commit
50195a6
·
verified ·
1 Parent(s): 5f2bd1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ from starlette.responses import RedirectResponse
85
  app = FastAPI()
86
 
87
  # Initialize AI Models
88
- qa_model = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
89
  image_captioning = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
90
  reader = easyocr.Reader(['en', 'fr']) # OCR for English & French
91
 
 
85
  app = FastAPI()
86
 
87
  # Initialize AI Models
88
+ qa_model = pipeline("question-answering", model="deepset/roberta-base-squad2")
89
  image_captioning = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
90
  reader = easyocr.Reader(['en', 'fr']) # OCR for English & French
91