xinah3131 commited on
Commit
60ef30a
·
1 Parent(s): b5373c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ def get_picture_from_url(url):
91
  # Function to make predictions
92
  def predict_trend(title, duration, category_id):
93
  duration = str(duration)
94
- category_id = str(category_id)
95
  clean_new_title = preprocess(title)
96
  # Join the preprocessed words back into a string
97
  clean_new_title_str = ' '.join(clean_new_title)
 
91
  # Function to make predictions
92
  def predict_trend(title, duration, category_id):
93
  duration = str(duration)
94
+ category_id = int(category_id)
95
  clean_new_title = preprocess(title)
96
  # Join the preprocessed words back into a string
97
  clean_new_title_str = ' '.join(clean_new_title)