arjunanand13 commited on
Commit
e470fae
·
verified ·
1 Parent(s): 410ae76

Update handler.py

Browse files
Files changed (1) hide show
  1. 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/florence-enphaseall2-25e"
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)