Refactored Predictor class and removed main execution
Browse files- predict.py +0 -7
predict.py
CHANGED
@@ -54,10 +54,3 @@ class Predictor(BasePredictor):
|
|
54 |
file_path = Path(tempfile.mkdtemp()) / file_name
|
55 |
mesh_render.save_rendering(str(file_path), output)
|
56 |
return file_path
|
57 |
-
|
58 |
-
|
59 |
-
if __name__ == "__main__":
|
60 |
-
predictor = Predictor()
|
61 |
-
predictor.setup()
|
62 |
-
|
63 |
-
print(predictor.predict("cat", 1, 0.0))
|
|
|
54 |
file_path = Path(tempfile.mkdtemp()) / file_name
|
55 |
mesh_render.save_rendering(str(file_path), output)
|
56 |
return file_path
|
|
|
|
|
|
|
|
|
|
|
|
|
|