fxn 0.0.15__py3-none-any.whl → 0.0.16__py3-none-any.whl

fxn/cli/predict.py

@@ -9,6 +9,7 @@ from numpy import ndarray
 from pathlib import Path, PurePath
 from PIL import Image
 from rich import print_json
+from rich.progress import Progress, SpinnerColumn, TextColumn
 from tempfile import mkstemp
 from typer import Argument, Context, Option
 
@@ -21,14 +22,20 @@ def predict (
     context: Context = 0
 ):
     # Predict
-    inputs = { context.args[i].replace("-", ""): _parse_value(context.args[i+1]) for i in range(0, len(context.args), 2) }
-    prediction = Prediction.create(
-        tag=tag,
-        **inputs,
-        raw_outputs=raw_outputs,
-        return_binary_path=True,
-        access_key=get_access_key()
-    )
+    with Progress(
+        SpinnerColumn(spinner_name="point"),
+        TextColumn("[progress.description]{task.description}"),
+        transient=True
+    ) as progress:
+        progress.add_task(description="Making prediction...", total=None)
+        inputs = { context.args[i].replace("-", ""): _parse_value(context.args[i+1]) for i in range(0, len(context.args), 2) }
+        prediction = Prediction.create(
+            tag=tag,
+            **inputs,
+            raw_outputs=raw_outputs,
+            return_binary_path=True,
+            access_key=get_access_key()
+        )
     # Parse results
     images = []
     if hasattr(prediction, "results") and prediction.results is not None:

fxn/version.py

@@ -3,4 +3,4 @@
 #   Copyright © 2023 NatML Inc. All Rights Reserved.
 #
 
-__version__ = "0.0.15"
+__version__ = "0.0.16"

{fxn-0.0.15.dist-info → fxn-0.0.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fxn
-Version: 0.0.15
+Version: 0.0.16
 Summary: Run on-device and cloud AI prediction functions in Python. Register at https://fxn.ai.
 Home-page: https://fxn.ai
 Author: NatML Inc.
@@ -33,33 +33,35 @@ Run AI prediction functions (a.k.a "predictors") in your Python apps. With Funct
 ## Installing Function
 Function is distributed on PyPi. This distribution contains both the Python client and the command line interface (CLI). To install, open a terminal and run the following command:
 ```sh
-pip install fxn
+pip install --upgrade fxn
 ```
 
 > Note that Function requires Python 3.9+
 
-## Signing in to Function
-Head over to [fxn.ai](https://fxn.ai) to create an account by logging in. Once you do, generate an access key:
-
-![generate access key](https://raw.githubusercontent.com/fxnai/.github/main/access_key.gif)
-
-Now, let's login to the Function CLI with your access key. Open a terminal and run the following command:
-```sh
-fxn auth login <ACCESS KEY>
+## Making a Prediction
+Let's run the [`@samplefxn/stable-diffusion`](https://fxn.ai/@samplefxn/stable-diffusion) predictor which accepts a text `prompt` and generates a corresponding image.
+
+### In Python
+Run the following Python script:
+```py
+import fxn
+
+prediction = fxn.Prediction.create(
+    tag="@samplefxn/stable-diffusion",
+    prompt="An astronaut riding a horse on Mars"
+)
+image = prediction.results[0]
+image.show()
 ```
 
-You should see information about your Function account:
-
-![login to CLI](https://raw.githubusercontent.com/fxnai/.github/main/auth_login.gif)
-
-## Making a Prediction
-You can start off by running an existing predictor [on Function](https://fxn.ai/explore). Let's run the [`@natml/stable-diffusion`](https://fxn.ai/@natml/stable-diffusion) predictor which accepts a text `prompt` and generates a corresponding image. In terminal, run the following command:
+### In the CLI
+Open up a terminal and run the following command:
 
 ```sh
-fxn predict @natml/stable-diffusion --prompt "An astronaut riding a horse on the moon"
+fxn predict @samplefxn/stable-diffusion --prompt "An astronaut riding a horse on the moon"
 ```
 
-And within a few seconds, you should see a creepy-looking image pop up 😅:
+Within a few seconds, you should see a creepy-looking image pop up 😅:
 
 ![prediction](https://raw.githubusercontent.com/fxnai/.github/main/predict.gif)
 

{fxn-0.0.15.dist-info → fxn-0.0.16.dist-info}/RECORD

@@ -1,6 +1,6 @@
 fxn/__init__.py,sha256=0NzZl5hRQ0MPNN4Wp9-ghqak2NJzhfHvVyBGX-IntKs,320
 fxn/magic.py,sha256=TRgE68OAo1djmeQRiqxnSJqa72td6zxnwo0s0bkdUKM,1041
-fxn/version.py,sha256=3OK-QR3joAr7h4-0cJiXXXmjS7nKU-UrxqfaU1n-DqU,95
+fxn/version.py,sha256=PVMyxP-O98TEP8WlKBEi04kzavQs4NlWxAjlIIBJ86w,95
 fxn/api/__init__.py,sha256=UcCERjadfSbqnt7urXYbrdRRFwA0QwpC9uBQYPJC5rE,460
 fxn/api/api.py,sha256=X2LRjer0s7ifecYsZmdqv5txEwPjA2ekdSCGUaQJ5So,927
 fxn/api/dtype.py,sha256=z9zjjsOk8X8LTCEpwLuWOTUys6nP_3L6VmZXLt0ilAY,617
@@ -16,11 +16,11 @@ fxn/cli/__init__.py,sha256=5-yqKCUJCiNTA_7EHfCj5GZKthIQSdbPcVAdMnzYNLg,1492
 fxn/cli/auth.py,sha256=_5rZ9L-blD39U6-SgHV9V2ltyjJ7k32NY3bLvoGsaNc,1659
 fxn/cli/env.py,sha256=mnyaL27n59pCU-p2Jm1fo03se7btMV2zdoJWWVshF_M,1658
 fxn/cli/misc.py,sha256=gLEGGWyawUQaLOuy8bfS2wrJuS-tN-bleOrv5A07xck,690
-fxn/cli/predict.py,sha256=Tfo-weSusIka33NQf8-iI3liH7RfYgs1195v2ikjI8s,2842
+fxn/cli/predict.py,sha256=QPNmwLAF8KndaVSS2vt0OBB-UoRn0VHgM2CNmIN6EEw,3180
 fxn/cli/predictors.py,sha256=vAL1FioQSseEiu7jbXY0DRiHnnMmThK2csJLThizS7A,3481
-fxn-0.0.15.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-fxn-0.0.15.dist-info/METADATA,sha256=FOIhE1rmbTUOrKQRjkeYdc0UU33XFUVAD8J1ds6Dy00,3213
-fxn-0.0.15.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-fxn-0.0.15.dist-info/entry_points.txt,sha256=QBwKIRed76CRY98VYQYrQDVEBZtJugxJJmBpilxuios,46
-fxn-0.0.15.dist-info/top_level.txt,sha256=1ULIEGrnMlhId8nYAkjmRn9g3KEFuHKboq193SEKQkA,4
-fxn-0.0.15.dist-info/RECORD,,
+fxn-0.0.16.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+fxn-0.0.16.dist-info/METADATA,sha256=guWZoXVbaAkQGXgtuCLx6UJ5tL4wCf4kzMpQDPs3GYo,2887
+fxn-0.0.16.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+fxn-0.0.16.dist-info/entry_points.txt,sha256=QBwKIRed76CRY98VYQYrQDVEBZtJugxJJmBpilxuios,46
+fxn-0.0.16.dist-info/top_level.txt,sha256=1ULIEGrnMlhId8nYAkjmRn9g3KEFuHKboq193SEKQkA,4
+fxn-0.0.16.dist-info/RECORD,,