Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -85,7 +85,8 @@ from starlette.responses import RedirectResponse
|
|
85 |
app = FastAPI()
|
86 |
|
87 |
# Initialize AI Models
|
88 |
-
qa_model = pipeline("
|
|
|
89 |
image_captioning = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
|
90 |
reader = easyocr.Reader(['en', 'fr']) # EasyOCR for image text extraction (English & French)
|
91 |
|
|
|
85 |
app = FastAPI()
|
86 |
|
87 |
# Initialize AI Models
|
88 |
+
qa_model = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
|
89 |
+
|
90 |
image_captioning = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
|
91 |
reader = easyocr.Reader(['en', 'fr']) # EasyOCR for image text extraction (English & French)
|
92 |
|