daviddoji/Made-With-ML
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

adding dotenv for credentials management

Browse Source
This commit is contained in:
Goku Mohandas
2023-09-15 09:44:38 -07:00
parent 0133795963
commit 097ffaadc8
9 changed files with 1713 additions and 1792 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
madewithml/predict.py
View File

@@ -125,7 +125,7 @@ def predict(
# Load components
best_checkpoint = get_best_checkpoint(run_id=run_id)
predictor = TorchPredictor.from_checkpoint(best_checkpoint)
preprocessor = predictor.get_preprocessor()
# preprocessor = predictor.get_preprocessor()
# Predict
sample_df = pd.DataFrame([{"title": title, "description": description, "tag": "other"}])