ikraamkb commited on
Commit
50fbe1a
Β·
verified Β·
1 Parent(s): b1622cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ import easyocr
17
  app = FastAPI()
18
 
19
  # Load AI Model for Question Answering on Documents (Mistral-7B)
20
- model_name = "TinyLlama/TinyLlama-1.1B"
21
  print(f"πŸ”„ Loading model: {model_name}...")
22
  tokenizer = AutoTokenizer.from_pretrained(model_name)
23
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
17
  app = FastAPI()
18
 
19
  # Load AI Model for Question Answering on Documents (Mistral-7B)
20
+ model_name = "meta-llama/Llama-2-7b-chat-hf"
21
  print(f"πŸ”„ Loading model: {model_name}...")
22
  tokenizer = AutoTokenizer.from_pretrained(model_name)
23
  model = AutoModelForCausalLM.from_pretrained(model_name)