Merge branch 'master' into issue_430

.pre-commit-config.yaml

@@ -23,7 +23,7 @@ repos:
       - id: end-of-file-fixer
 
   - repo: https://github.com/PyCQA/isort
-    rev: 5.8.0
+    rev: 5.9.1
     hooks:
       - id: isort
 

README.md

@@ -119,12 +119,12 @@ print(predictions)
 
 ### Serving
 
-`Serve` is a framework agnostic serving engine ! [Learn more](https://lightning-flash.readthedocs.io/en/latest/general/serve.html#) and [find examples](flash_examples/serve/generic/boston_prediction/inference_server.py).
+`Serve` is a framework agnostic serving engine ! [Learn more](https://lightning-flash.readthedocs.io/en/latest/general/serve.html#) and [check out our examples](flash_examples/serve).
 
 ```python
-from flash.text import TranslationTask
+from flash.text import TextClassifier
 
-model = TranslationTask.load_from_checkpoint("https://flash-weights.s3.amazonaws.com/translation_model_en_ro.pt")
+model = TextClassifier.load_from_checkpoint("https://flash-weights.s3.amazonaws.com/text_classification_model.pt")
 model.serve()
 ```
 

flash/core/model.py

@@ -91,9 +91,11 @@ def __new__(mcs, *args, **kwargs):
         result = ABCMeta.__new__(mcs, *args, **kwargs)
         if result.required_extras is not None:
             result.__init__ = _requires_extras(result.required_extras)(result.__init__)
-            result.load_from_checkpoint = classmethod(
-                _requires_extras(result.required_extras)(result.load_from_checkpoint.__func__)
-            )
+            load_from_checkpoint = getattr(result, "load_from_checkpoint", None)
+            if load_from_checkpoint is not None:
+                result.load_from_checkpoint = classmethod(
+                    _requires_extras(result.required_extras)(result.load_from_checkpoint.__func__)
+                )
         return result