handler.py modification
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -9,7 +9,7 @@ from io import BytesIO
|
|
9 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
10 |
|
11 |
class EndpointHandler:
|
12 |
-
def __init__(self, path: str = "morthens/qwen2-vl-
|
13 |
# Load the processor and model
|
14 |
self.processor = AutoProcessor.from_pretrained(path)
|
15 |
self.model = Qwen2VLForConditionalGeneration.from_pretrained(
|
|
|
9 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
10 |
|
11 |
class EndpointHandler:
|
12 |
+
def __init__(self, path: str = "morthens/qwen2-vl-infer"):
|
13 |
# Load the processor and model
|
14 |
self.processor = AutoProcessor.from_pretrained(path)
|
15 |
self.model = Qwen2VLForConditionalGeneration.from_pretrained(
|