wealthcoders commited on
Commit
f2d4b69
·
verified ·
1 Parent(s): 17b899c

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -19,7 +19,7 @@ prompt = """**Task**:
19
  **Important**: the output format must be Markdown (use bold text, headlines, so on)."""
20
 
21
  class EndpointHandler:
22
- def __init__(self, path: str = "Qwen/Qwen3-VL-2B-Instruct"):
23
  # Load tokenizer and model
24
  self.processor = AutoProcessor.from_pretrained(path)
25
  self.model = Qwen3VLForConditionalGeneration.from_pretrained(path, device_map="auto")
 
19
  **Important**: the output format must be Markdown (use bold text, headlines, so on)."""
20
 
21
  class EndpointHandler:
22
+ def __init__(self, path: str = "unsloth/Qwen3-VL-2B-Instruct-unsloth-bnb-4bit"):
23
  # Load tokenizer and model
24
  self.processor = AutoProcessor.from_pretrained(path)
25
  self.model = Qwen3VLForConditionalGeneration.from_pretrained(path, device_map="auto")