Update app.py

app.py

@@ -106,9 +106,27 @@ start = st.text_input("Beginning of the song:", "But for me to rap like a comput
 TOKEN = "q_JK_BFy9OMiG7fGTzL-nUto9JDv3iXI24aYRrQnkOvjSCSbY4BuFIindweRsr5I"
 genius = lyricsgenius.Genius(TOKEN)
 
+model_html = """
+
+<div class="inline-flex flex-col" style="line-height: 1.5;">
+    <div class="flex">
+        <div
+			style="display:DISPLAY_1; margin-left: auto; margin-right: auto; width: 92px; height:92px; border-radius: 50%; background-size: cover; background-image: url(&#39;USER_PROFILE&#39;)">
+        </div>
+    </div>
+    <div style="text-align: center; margin-top: 3px; font-size: 16px; font-weight: 800">🤖 HuggingArtists Model 🤖</div>
+    <div style="text-align: center; font-size: 16px; font-weight: 800">USER_NAME</div>
+    <a href="https://genius.com/artists/USER_HANDLE">
+    	<div style="text-align: center; font-size: 14px;">@USER_HANDLE</div>
+    </a>
+</div>
+"""
 
 if st.button("Run"):
     with st.spinner(text=f"Downloading the model by {artist_name }..."):
         artist = genius.search_artist(artist_name, max_songs=0, get_full_info=False)
         if artist is not None:
-            st.image(artist.image_url, use_column_width=True)
+            artist_dict = genius.artist(artist.id)['artist']
+            artist_url = str(artist_dict['url'])
+            model_name = artist_url[artist_url.rfind('/') + 1:].lower()
+            st.markdown(model_html.format(USER_PROFILE=artist.image_url, USER_NAME=artist.name, USER_HANDLE=model_name ))