Update app.py

app.py

@@ -82,16 +82,12 @@ def process_image_tag(image_tag, repo_id, oauth_token: Union[gr.OAuthToken, None
         # Delete the blobs folder
         shutil.rmtree('blobs')
 
-        return f'Find your repo <a href=\'{repo_path}\' target="_blank" style="text-decoration:underline">here</a>'
+        return (f'Find your repo <a href=\'{repo_path}\' target="_blank" style="text-decoration:underline">here</a>', "dramallama.jpg")
         
     except Exception as e:
         shutil.rmtree('blobs', ignore_errors=True)
 
-        return f"We got an error, my dude, here's what the error looks like: {str(e)}"
-
-css = """
-.main_ui_logged_out{opacity: 0.3; pointer-events: none}
-"""
+        return (f"We got an error, my dude, here's what the error looks like: {str(e)}", "madllama.jpg")
 
 # Create the Gradio interface using gr.Blocks
 with gr.Blocks() as demo:
@@ -106,7 +102,7 @@ with gr.Blocks() as demo:
     result_output = gr.Markdown(label="Result")
     
     process_button = gr.Button("Process")
-    process_button.click(fn=process_image_tag, inputs=[image_tag_input, repo_id_input], outputs=result_output)
+    process_button.click(fn=process_image_tag, inputs=[image_tag_input, repo_id_input], outputs=[result_output, gr.Image(show_label=False)])
 
 # Launch the Gradio app
 demo.launch()