Initial commit.
This commit is contained in:
BIN
helpers/__pycache__/ai_helper.cpython-310.pyc
Normal file
BIN
helpers/__pycache__/ai_helper.cpython-310.pyc
Normal file
Binary file not shown.
4
helpers/ai_helper.py
Normal file
4
helpers/ai_helper.py
Normal file
@@ -0,0 +1,4 @@
|
||||
class Ai_Helper:
|
||||
@staticmethod
|
||||
def get_pretrained_model_path(model_name, is_fine_tuned = False):
|
||||
return f"./pretrained_models/{model_name}{'-finetuned' if is_fine_tuned else ''}"
|
||||
Reference in New Issue
Block a user