wiusdy commited on
Commit
8c94c5c
1 Parent(s): 80ba3ac

testing the selected model

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -14,7 +14,7 @@ class Inference:
14
  def inference(self, selected, image, text):
15
  self.logger.info(f"selected model {selected}")
16
  if selected == "Model 1":
17
- return self.__inference_deplot(image, text)
18
  elif selected == "Model 2":
19
  return self.__inference_deplot(image, text)
20
  elif selected == "Model 3":
 
14
  def inference(self, selected, image, text):
15
  self.logger.info(f"selected model {selected}")
16
  if selected == "Model 1":
17
+ return self.__inference_vilt(image, text)
18
  elif selected == "Model 2":
19
  return self.__inference_deplot(image, text)
20
  elif selected == "Model 3":