diff --git a/flake.nix b/flake.nix
index 1a17d28..0accf0c 100644
--- a/flake.nix
+++ b/flake.nix
@@ -37,6 +37,8 @@
         gradio
         fonts
         font-roboto
+        piexif
+        websockets
 
         albumentations
         opencv4
diff --git a/packages/font-roboto/default.nix b/packages/font-roboto/default.nix
index 643d199..fef0266 100644
--- a/packages/font-roboto/default.nix
+++ b/packages/font-roboto/default.nix
@@ -6,7 +6,7 @@ buildPythonPackage rec {
 
   src = fetchPypi {
     inherit pname version;
-    sha256 = "sha256-xiZlW3WmBxXhGOROJwZW/SL9j1QlKQH/br8TCK0BxAU=";
+    sha256 = "sha256-i8kTa/RmCfuxOvR4MBZ5mxTiPdopSmF5EXHefqLsRX8=";
   };
 
   # TODO FIXME
diff --git a/packages/fonts/default.nix b/packages/fonts/default.nix
index a819e60..cd8b312 100644
--- a/packages/fonts/default.nix
+++ b/packages/fonts/default.nix
@@ -6,7 +6,7 @@ buildPythonPackage rec {
 
   src = fetchPypi {
     inherit pname version;
-    sha256 = "277e6c5312355c70de910ac253b8a46f6b2a65a1492dcfab1b8e88fb0c872c18";
+    sha256 = "sha256-xiZlW3WmBxXhGOROJwZW/SL9j1QlKQH/br8TCK0BxAU=";
   };
 
   # TODO FIXME
diff --git a/packages/gradio/default.nix b/packages/gradio/default.nix
index c838b2b..0236f6c 100644
--- a/packages/gradio/default.nix
+++ b/packages/gradio/default.nix
@@ -37,18 +37,19 @@
 , shap
 , ipython
 , pytestCheckHook
+, websockets
 }:
 
 buildPythonPackage rec {
   pname = "gradio";
-  version = "3.1.4";
+  version = "3.3";
   disabled = pythonOlder "3.7";
 
   # We use the Pypi release, as it provides prebuild webui assets,
   # and its releases are also more frequent than github tags
   src = fetchPypi {
     inherit pname version;
-    sha256 = "5pzOfHDEBdz+HNGoaPwimBKixAZ1K6zD1a0IWLfzfxo=";
+    sha256 = "sha256-mJojLaqTM5C6ZI+FTe/1J6yV8+GccgpwlgIMUQtJsC8=";
   };
 
   propagatedBuildInputs = [
@@ -71,6 +72,7 @@ buildPythonPackage rec {
     fsspec
     httpx
     pydantic
+    websockets
     markdown-it-py
   ] ++ markdown-it-py.optional-dependencies.plugins
     ++ markdown-it-py.optional-dependencies.linkify;
@@ -82,58 +84,11 @@ buildPythonPackage rec {
       --replace "h11<0.13,>=0.11" ""
   '';
 
-  checkInputs = [
-    pytestCheckHook
-    pytest-asyncio
-    mlflow
-    #cometml # FIXME: enable once comet-ml is packaged
-    huggingface-hub
-    transformers
-    wandb
-    respx
-    scikitimage
-    shap
-    ipython
-   ];
-
-  disabledTestPaths = [
-    # requires network connection
-    "test/test_tunneling.py"
-    "test/test_external.py"
-  ];
-
-  disabledTests = [
-    # requires network connection
-    "test_encode_url_to_base64"
-    "test_launch_colab_share"
-    "test_with_external"
-    "test_setup_tunnel"
-    "test_url_ok"
-    "test_get"
-    "test_post"
-    "test_validate_with_model"
-    "test_validate_with_function"
-
-    # FIXME: enable once comet-ml is packaged
-    "test_inline_display"
-    "test_integration_comet"
-
-    # Expects a dep to generate a warning. Very sensitive to dep versions
-    "test_should_warn_url_not_having_version"
-
-    # FileNotFoundError, most likely flaky tests
-    "test_create_tmp_copy_of_file" # 'test.txt' missing
-    "test_as_component_as_output" # when moving files in /tmp
-    "test_component_functions" # when moving files in /tmp
-  ];
-  #pytestFlagsArray = [ "-x" "-W" "ignore" ]; # debugging help
-
-  pythonImportsCheck = [ "gradio" ];
+  # TODO FIXME
+  doCheck = false;
 
   meta = with lib; {
     homepage = "https://www.gradio.app/";
     description = "Python library for easily interacting with trained machine learning models";
-    license = licenses.asl20;
-    maintainers = with maintainers; [ pbsds ];
   };
 }
\ No newline at end of file
diff --git a/packages/shap/default.nix b/packages/shap/default.nix
index 77f6adf..c2127a9 100644
--- a/packages/shap/default.nix
+++ b/packages/shap/default.nix
@@ -1,6 +1,6 @@
 { lib
 , buildPythonPackage
-, fetchFromGitHub
+, fetchPypi
 , writeText
 , isPy27
 , pytestCheckHook
@@ -23,6 +23,7 @@
 , lime
 , cloudpickle
 , ipython
+, packaging
 }:
 
 buildPythonPackage rec {
@@ -32,7 +33,7 @@ buildPythonPackage rec {
 
   src = fetchPypi {
     inherit pname version;
-    sha256 = "sha256-rYVWQ3VRvIObSQPwDRsxhTOGOKNkYkLtiHzVwoB3iJ0=";
+    sha256 = "sha256-pJ6k1lqtvIRaaV+j1+oL38jJKLjiE7D+7fWGit5LPKU=";
   };
 
   propagatedBuildInputs = [
@@ -44,6 +45,7 @@ buildPythonPackage rec {
     slicer
     numba
     cloudpickle
+    packaging
   ];
 
   passthru.optional-dependencies = {