sree4411 commited on
Commit
6fbaf6c
Β·
verified Β·
1 Parent(s): 1a22490

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -8,8 +8,7 @@ def custom_tokenizer(text):
8
  # Modify this function to match your original tokenizer logic
9
  return text.lower().split()
10
 
11
- # 🧠 Debug: Show current directory contents to verify files
12
- st.write("πŸ“‚ Files in current directory:", os.listdir())
13
 
14
  # πŸ”ƒ Load model files
15
  try:
@@ -36,7 +35,7 @@ def predict_tags(title, description):
36
  input_vector = vectorizer.transform([input_text])
37
  prediction = model.predict(input_vector)
38
  predicted_tags = mlb.inverse_transform(prediction)
39
-
40
  if predicted_tags and predicted_tags[0]:
41
  return "βœ… Predicted Tags: " + ", ".join(predicted_tags[0])
42
  else:
 
8
  # Modify this function to match your original tokenizer logic
9
  return text.lower().split()
10
 
11
+
 
12
 
13
  # πŸ”ƒ Load model files
14
  try:
 
35
  input_vector = vectorizer.transform([input_text])
36
  prediction = model.predict(input_vector)
37
  predicted_tags = mlb.inverse_transform(prediction)
38
+ st.write(predicted_tags)
39
  if predicted_tags and predicted_tags[0]:
40
  return "βœ… Predicted Tags: " + ", ".join(predicted_tags[0])
41
  else: