ikraamkb commited on
Commit
875917f
Β·
verified Β·
1 Parent(s): c77309c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ app = FastAPI()
18
 
19
  print(f"πŸ”„ Loading models")
20
 
21
- doc_qa_pipeline = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-chat")
22
  image_captioning_pipeline = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
23
  print("models loaded")
24
  # Initialize OCR Model (Lazy Load)
 
18
 
19
  print(f"πŸ”„ Loading models")
20
 
21
+ doc_qa_pipeline = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0")
22
  image_captioning_pipeline = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
23
  print("models loaded")
24
  # Initialize OCR Model (Lazy Load)