vikhyatk
/

moondream2

Image-Text-to-Text

text-generation

Model card Files Files and versions

zesquirrelnator commited on May 22, 2024

Commit

43b595d

·

verified ·

1 Parent(s): aa98b0a

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -6,7 +6,7 @@ import base64
 class EndpointHandler:
     def __init__(self, model_dir):
-        self.model_id = "zesquirrelnator/moondream2-finetuneV2"
         self.model = AutoModelForCausalLM.from_pretrained(self.model_id, trust_remote_code=True)
         self.tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2", trust_remote_code=True)

 class EndpointHandler:
     def __init__(self, model_dir):
+        self.model_id = "vikhyatk/moondream2"
         self.model = AutoModelForCausalLM.from_pretrained(self.model_id, trust_remote_code=True)
         self.tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2", trust_remote_code=True)