ikraamkb commited on
Commit
b34af62
·
verified ·
1 Parent(s): ae8b0f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -85,7 +85,8 @@ from starlette.responses import RedirectResponse
85
  app = FastAPI()
86
 
87
  # Initialize AI Models
88
- qa_model = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0")
 
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