zesquirrelnator commited on
Commit
25cea24
·
verified ·
1 Parent(s): cee3dee

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -5,9 +5,9 @@ from io import BytesIO
5
  import base64
6
 
7
  # Initialize the model and tokenizer
8
- model_id = "vikhyatk/moondream2"
9
  model = AutoModelForCausalLM.from_pretrained(model_id)
10
- tokenizer = AutoTokenizer.from_pretrained(model_id)
11
 
12
  # Check if CUDA (GPU support) is available and then set the device to GPU or CPU
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
5
  import base64
6
 
7
  # Initialize the model and tokenizer
8
+ model_id = "zesquirrelnator/moondream2-finetuneV2"
9
  model = AutoModelForCausalLM.from_pretrained(model_id)
10
+ tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
11
 
12
  # Check if CUDA (GPU support) is available and then set the device to GPU or CPU
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")