BoghdadyJR commited on
Commit
e935d19
·
verified ·
1 Parent(s): a217da8

Add inference handler for HF Endpoints

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -7,8 +7,8 @@ import base64
7
  from peft import PeftModel
8
 
9
  class EndpointHandler():
10
- def __init__(self):
11
- self.path = "BoghdadyJR/Qwen_UI_final"
12
  # Load base model and tokenizer
13
  base_model_id = "Qwen/Qwen2-VL-2B-Instruct"
14
 
 
7
  from peft import PeftModel
8
 
9
  class EndpointHandler():
10
+ def __init__(self, model_dir: str):
11
+ self.path = model_dir
12
  # Load base model and tokenizer
13
  base_model_id = "Qwen/Qwen2-VL-2B-Instruct"
14