ikraamkb commited on
Commit
463b3e2
·
verified ·
1 Parent(s): 50195a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -11,6 +11,8 @@ import fitz # PyMuPDF for PDF
11
 
12
  app = FastAPI()
13
 
 
 
14
  # ========== Document QA Setup ==========
15
  doc_tokenizer = AutoTokenizer.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
16
  doc_model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
 
11
 
12
  app = FastAPI()
13
 
14
+
15
+
16
  # ========== Document QA Setup ==========
17
  doc_tokenizer = AutoTokenizer.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
18
  doc_model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")