Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ team_encoder = joblib.load("team_encoder.pkl")
|
|
| 14 |
|
| 15 |
# Load the model with full object support
|
| 16 |
multi_label_classifier = torch.load("multi_label_classifier.pth", weights_only=False)
|
| 17 |
-
multi_label_classifier.eval()
|
| 18 |
|
| 19 |
# Dummy function to get keywords
|
| 20 |
def get_top_keywords_per_category(category, n=5):
|
|
|
|
| 14 |
|
| 15 |
# Load the model with full object support
|
| 16 |
multi_label_classifier = torch.load("multi_label_classifier.pth", weights_only=False)
|
| 17 |
+
# multi_label_classifier.eval()
|
| 18 |
|
| 19 |
# Dummy function to get keywords
|
| 20 |
def get_top_keywords_per_category(category, n=5):
|