Upload app.py

app.py

@@ -19,7 +19,7 @@ with h5py.File('complete_artist_data.hdf5', 'r') as f:
     # Load artist names and decode to strings
     artist_names = [name.decode() for name in f['artist_names'][:]]
 
-def find_similar_artists(new_tags_string):
+def find_similar_artists(new_tags_string, top_n):
     new_image_tags = [tag.strip() for tag in new_tags_string.split(",")]
     unseen_tags = set(new_image_tags) - set(vectorizer.vocabulary_.keys())
     unseen_tags_str = f'Unseen Tags: {", ".join(unseen_tags)}' if unseen_tags else 'No unseen tags.'
@@ -27,7 +27,6 @@ def find_similar_artists(new_tags_string):
     X_new_image = vectorizer.transform([','.join(new_image_tags)])
     similarities = cosine_similarity(X_new_image, X_artist)[0]
     
-    top_n = 20
     top_artist_indices = np.argsort(similarities)[-top_n:][::-1]
     bottom_artist_indices = np.argsort(similarities)[:top_n]
     
@@ -42,10 +41,13 @@ def find_similar_artists(new_tags_string):
 
 iface = gr.Interface(
     fn=find_similar_artists,
-    inputs="text",
+    inputs=[
+        gr.Textbox(label="Enter image tags", placeholder="fox, outside, detailed background"),
+        gr.Slider(minimum=1, maximum=100, default=10, step=1, label="Number of artists")
+    ],
     outputs="text",
-    title="Artist Similarity Finder",
-    description="Enter image tags to find similar artists based on learned similarities."
+    title="Tagset Completer",
+    description="Enter a list of comma-separated e6 tags"
 )
 
 iface.launch()