awacke1 commited on
Commit
574bc1a
1 Parent(s): 00e56d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -107,6 +107,11 @@ def load_example_files():
107
  return None
108
 
109
  def cluster_sentences(sentences, num_clusters):
 
 
 
 
 
110
  # Vectorize the sentences
111
  vectorizer = TfidfVectorizer()
112
  X = vectorizer.fit_transform(sentences)
 
107
  return None
108
 
109
  def cluster_sentences(sentences, num_clusters):
110
+ # Check if the number of sentences is less than the desired number of clusters
111
+ if len(sentences) < num_clusters:
112
+ # If so, adjust the number of clusters to match the number of sentences
113
+ num_clusters = len(sentences)
114
+
115
  # Vectorize the sentences
116
  vectorizer = TfidfVectorizer()
117
  X = vectorizer.fit_transform(sentences)