Spaces:

ikraamkb
/

qtAnswering

Running

ikraamkb commited on Mar 24

Commit

683cb5c

verified ·

1 Parent(s): 1be9899

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ app = FastAPI()
 print(f"🔄 Loading models")
-doc_qa_pipeline = pipeline("text2text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", device=-1)
 image_captioning_pipeline = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
 print("✅ Models loaded")

 print(f"🔄 Loading models")
+doc_qa_pipeline = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", device=-1)
 image_captioning_pipeline = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
 print("✅ Models loaded")