Ishaan Gupta
commited on
Commit
•
8009a47
1
Parent(s):
0361a01
updated to model path in handler
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -10,8 +10,8 @@ class EndpointHandler():
|
|
10 |
# Preload all the elements you are going to need at inference.
|
11 |
# pseudo:
|
12 |
# self.model= load_model(path)
|
13 |
-
self.model = AutoModelForVision2Seq.from_pretrained(
|
14 |
-
self.processor = AutoProcessor.from_pretrained(
|
15 |
|
16 |
# prompt = "<grounding>An image of"
|
17 |
|
|
|
10 |
# Preload all the elements you are going to need at inference.
|
11 |
# pseudo:
|
12 |
# self.model= load_model(path)
|
13 |
+
self.model = AutoModelForVision2Seq.from_pretrained(path).to("cuda")
|
14 |
+
self.processor = AutoProcessor.from_pretrained(path)
|
15 |
|
16 |
# prompt = "<grounding>An image of"
|
17 |
|