Upload 2 files

app.py

@@ -204,7 +204,6 @@ with open("word_rating_probabilities.csv", 'r', newline='', encoding='utf-8') as
             nsfw_tags.add(word)
     
     
-    
 sample_images_directory_path = 'sampleimages'
 def generate_artist_image_tuples(top_artists, image_directory):
     json_files = glob.glob(f'{image_directory}/*.json')
@@ -655,11 +654,12 @@ with gr.Blocks(css=css) as app:
                 image_tags = gr.Textbox(label="Enter Prompt", placeholder="e.g. fox, outside, detailed background, ...")
                 bad_tags_illustrated_string = gr.HighlightedText(show_legend=True, color_map={"Unknown Tag":"red","Duplicate":"yellow","Remove Final Comma":"purple","Move Comma Inside Parentheses":"green"}, label="Annotated Prompt")
             with gr.Column(scale=1):
-                #gr.Image(label=" ", value="transparentsquirrel.png", height=155, width=140)
                 #image_path = os.path.join("https://huggingface.co/spaces/FoodDesert/Prompt_Squirrel/resolve/main", "transparentsquirrel.png")
-                #gr.HTML('<div style="text-align: center;"><img src="transparentsquirrel.png" alt="Cute Mascot" style="max-height: 100px; background: transparent;"></div><br>')
+                #gr.Image(label=" ", value=image_path, height=155, width=140)
+                image_path = os.path.join('mascotimages', "transparentsquirrel.png") 
+                #gr.HTML('<div style="text-align: center;"><img src={image_path} alt="Cute Mascot" style="max-height: 100px; background: transparent;"></div><br>')
                 #gr.HTML("<br>" * 2)  # Adjust the number of line breaks ("<br>") as needed to push the button down
-                gr.Image(value='./transparentsquirrel.png',show_label=False, show_download_button=False, show_share_button=False, height=200)
+                gr.Image(value=image_path,show_label=False, show_download_button=False, show_share_button=False, height=200)
                 submit_button = gr.Button(variant="primary")
     with gr.Row():
         with gr.Column(scale=3):