diff --git a/README.md b/README.md
index 3427759..47c6c41 100644
--- a/README.md
+++ b/README.md
@@ -23,7 +23,7 @@
 </p>
 
 
-<img width="1216" align="center" alt="whisperx-arch" src="figures/pipeline.png">
+<img width="1216" align="center" alt="whisperx-arch" src="https://raw.githubusercontent.com/m-bain/whisperX/refs/heads/main/figures/pipeline.png">
 
 
 <!-- <p align="left">Whisper-Based Automatic Speech Recognition (ASR) with improved timestamp accuracy + quality via forced phoneme alignment and voice-activity based batching for fast inference.</p> -->
diff --git a/setup.py b/setup.py
index 203a079..6bd6f67 100644
--- a/setup.py
+++ b/setup.py
@@ -1,15 +1,18 @@
 import os
-import platform
 
 import pkg_resources
 from setuptools import find_packages, setup
 
+with open("README.md") as f:
+    long_description = f.read()
+
 setup(
     name="whisperx",
     py_modules=["whisperx"],
     version="3.2.0",
     description="Time-Accurate Automatic Speech Recognition using Whisper.",
-    readme="README.md",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
     python_requires=">=3.9, <3.13",
     author="Max Bain",
     url="https://github.com/m-bain/whisperx",