kevin-pek commited on
Commit
7bdd3d6
·
1 Parent(s): 4b9dac6

change to small model

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -65,7 +65,7 @@ class StoppingCriteriaScores(StoppingCriteria):
65
  return all(self.stopped.values()) and len(self.stopped) > 0
66
 
67
  class EndpointHandler():
68
- def __init__(self, path="facebook/nougat-base") -> None:
69
  self.processor = NougatProcessor.from_pretrained(path)
70
  self.model = VisionEncoderDecoderModel.from_pretrained(path)
71
 
 
65
  return all(self.stopped.values()) and len(self.stopped) > 0
66
 
67
  class EndpointHandler():
68
+ def __init__(self, path="facebook/nougat-small") -> None:
69
  self.processor = NougatProcessor.from_pretrained(path)
70
  self.model = VisionEncoderDecoderModel.from_pretrained(path)
71