Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -27,7 +27,7 @@ class EndpointHandler:
|
|
| 27 |
print(f"Using device: {self.device}")
|
| 28 |
|
| 29 |
# Load the model
|
| 30 |
-
self.model_name = "arjunanand13/
|
| 31 |
self.model = AutoModelForCausalLM.from_pretrained(
|
| 32 |
self.model_name, trust_remote_code=True
|
| 33 |
).to(self.device)
|
|
|
|
| 27 |
print(f"Using device: {self.device}")
|
| 28 |
|
| 29 |
# Load the model
|
| 30 |
+
self.model_name = "arjunanand13/LADP_Florence-40e"
|
| 31 |
self.model = AutoModelForCausalLM.from_pretrained(
|
| 32 |
self.model_name, trust_remote_code=True
|
| 33 |
).to(self.device)
|