Upload app.py

app.py

@@ -226,7 +226,7 @@ def create_app():
     )
     def get_gpu_kind():
       device = jax.devices()[0]
-      if not gradio_helpers.should_mock() and device.platform != 'gpu':
+      if not gradio_helpers.should_mock():
         raise gr.Error('GPU not visible to JAX!')
       return f'GPU={device.device_kind}'
     demo.load(get_gpu_kind, None, gpu_kind)