Spaces:
Sleeping
Sleeping
testing the selected model
Browse files- 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.
|
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":
|