Spaces:
Running
Running
Update app.py
Browse files
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="
|
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 |
|